From f6af34626d37f63ecb128e1f775ebcbbc1d0e5bf Mon Sep 17 00:00:00 2001 From: Harry Mellor <19981378+hmellor@users.noreply.github.com> Date: Tue, 3 Feb 2026 12:07:24 +0000 Subject: [PATCH] Fix offline test for Transformers v5 (#33682) Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> --- tests/entrypoints/offline_mode/test_offline_mode.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/entrypoints/offline_mode/test_offline_mode.py b/tests/entrypoints/offline_mode/test_offline_mode.py index 539ff89ab..ad7a3960d 100644 --- a/tests/entrypoints/offline_mode/test_offline_mode.py +++ b/tests/entrypoints/offline_mode/test_offline_mode.py @@ -109,8 +109,15 @@ def _re_import_modules(): if k.startswith("transformers") and not k.startswith("transformers_modules") ] + # These modules are aliased in Transformers v5 and so cannot be reloaded directly + aliased_modules = ["tokenization_utils", "tokenization_utils_fast"] + reload_exception = None for module_name in hf_hub_module_names + transformers_module_names: + if any(module_name.endswith(f".{alias}") for alias in aliased_modules): + # Remove from sys.modules so they are re-aliased on next import + del sys.modules[module_name] + continue try: importlib.reload(sys.modules[module_name]) except Exception as e: