From 5f9f222c23e7a8409776c513f07828c1d545bef0 Mon Sep 17 00:00:00 2001 From: sambhavnoobcoder Date: Thu, 23 Jan 2025 22:05:15 +0530 Subject: [PATCH] ruff fix to test file --- tests/utils/test_import_utils.py | 23 +++++++++++------------ 1 file changed, 11 insertions(+), 12 deletions(-) diff --git a/tests/utils/test_import_utils.py b/tests/utils/test_import_utils.py index e26a9c3812b2..7e1883dc671d 100644 --- a/tests/utils/test_import_utils.py +++ b/tests/utils/test_import_utils.py @@ -1,30 +1,29 @@ import sys -import pytest -from transformers.utils.import_utils import clear_import_cache, _LazyModule + +from transformers.utils.import_utils import clear_import_cache + def test_clear_import_cache(): # Import some transformers modules - from transformers import AutoModel, AutoTokenizer - + # Get initial module count initial_modules = { - name: mod for name, mod in sys.modules.items() + name: mod for name, mod in sys.modules.items() if name.startswith('transformers.') } - + # Verify we have some modules loaded assert len(initial_modules) > 0 - + # Clear cache clear_import_cache() - + # Check modules were removed remaining_modules = { - name: mod for name, mod in sys.modules.items() + name: mod for name, mod in sys.modules.items() if name.startswith('transformers.') } assert len(remaining_modules) < len(initial_modules) - + # Verify we can reimport - from transformers import AutoModel, AutoTokenizer - assert 'transformers.models.auto.modeling_auto' in sys.modules \ No newline at end of file + assert 'transformers.models.auto.modeling_auto' in sys.modules