fixing

sambhavnoobcoder · Feb 3, 2025 · 75ed6de · 75ed6de
1 parent 388b2ac
commit 75ed6de
Showing 1 changed file with 4 additions and 11 deletions.
diff --git a/tests/utils/test_import_utils.py b/tests/utils/test_import_utils.py
@@ -2,15 +2,12 @@
 
 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()
-        if name.startswith('transformers.')
-    }
+    initial_modules = {name: mod for name, mod in sys.modules.items() if name.startswith("transformers.")}
 
     # Verify we have some modules loaded
     assert len(initial_modules) > 0
@@ -19,12 +16,8 @@ def test_clear_import_cache():
     clear_import_cache()
 
     # Check modules were removed
-    remaining_modules = {
-        name: mod for name, mod in sys.modules.items()
-        if name.startswith('transformers.')
-    }
+    remaining_modules = {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
+    assert "transformers.models.auto.modeling_auto" in sys.modules