fixup for test

sambhavnoobcoder · Jan 23, 2025 · 689b39f · 689b39f
1 parent 729939b
commit 689b39f
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/tests/utils/test_import_utils.py b/tests/utils/test_import_utils.py
@@ -7,10 +7,7 @@ def test_clear_import_cache():
     # Import some transformers modules
 
     # 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,11 +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
-    assert 'transformers.models.auto.modeling_auto' in sys.modules
+    assert "transformers.models.auto.modeling_auto" in sys.modules