From 689b39f6e3cd87f49322a6f9034d409f2c959c2c Mon Sep 17 00:00:00 2001 From: sambhavnoobcoder Date: Thu, 23 Jan 2025 22:09:11 +0530 Subject: [PATCH] fixup for test --- tests/utils/test_import_utils.py | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/tests/utils/test_import_utils.py b/tests/utils/test_import_utils.py index 7e1883dc671d..3d846174aca1 100644 --- 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