fix style changes

sambhavnoobcoder · Jan 23, 2025 · 3b4836c · 3b4836c
1 parent 33b7091
commit 3b4836c
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/utils/test_import_utils.py b/tests/utils/test_import_utils.py
@@ -1,14 +1,14 @@
 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.')
     }
 
@@ -20,11 +20,11 @@ def test_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 
+    assert 'transformers.models.auto.modeling_auto' in sys.modules