test check

sambhavnoobcoder · Jan 23, 2025 · 33b7091 · 33b7091
1 parent 393869b
commit 33b7091
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/tests/utils/test_import_utils.py b/tests/utils/test_import_utils.py
@@ -1,13 +1,16 @@
 import sys
-
-from transformers.utils.import_utils import clear_import_cache
-
+import pytest
+from transformers.utils.import_utils import clear_import_cache, _LazyModule
 
 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
@@ -16,8 +19,12 @@ 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
-    assert "transformers.models.auto.modeling_auto" in sys.modules
+    from transformers import AutoModel, AutoTokenizer
+    assert 'transformers.models.auto.modeling_auto' in sys.modules