From ba2910f73a0341b46525a3441da2e7cb66698d58 Mon Sep 17 00:00:00 2001 From: Harry Mellor <19981378+hmellor@users.noreply.github.com> Date: Wed, 25 Mar 2026 11:39:48 +0000 Subject: [PATCH] Fix offline mode test for Transformers v5 (#38095) Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> --- tests/entrypoints/offline_mode/test_offline_mode.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/tests/entrypoints/offline_mode/test_offline_mode.py b/tests/entrypoints/offline_mode/test_offline_mode.py index ad7a3960dd..55a3b98580 100644 --- a/tests/entrypoints/offline_mode/test_offline_mode.py +++ b/tests/entrypoints/offline_mode/test_offline_mode.py @@ -7,6 +7,7 @@ import importlib import sys import pytest +import regex as re import urllib3 from vllm import LLM @@ -110,11 +111,15 @@ def _re_import_modules(): ] # These modules are aliased in Transformers v5 and so cannot be reloaded directly - aliased_modules = ["tokenization_utils", "tokenization_utils_fast"] + aliased_module_patterns = [ + r".+\.tokenization_utils$", + r".+\.tokenization_utils_fast$", + r".+\.models\..+\.image_processing_.+_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): + if any(re.match(pattern, module_name) for pattern in aliased_module_patterns): # Remove from sys.modules so they are re-aliased on next import del sys.modules[module_name] continue