riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-20 04:58:22 +06:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Fix test isolation for clear_import_cache utility (#36345)

Browse Source 
* test fixup

* test fixup

* fixing tests for unused imports

* style fixes

* fix

* style fixes

* styke fix

* remove isolated module cache

* rm custom subprocess defination

* run using exsiting fn

* style fixup

* make fixup

* remove redundant comments

* rm redundat skipif + style changes
This commit is contained in:
Sambhav Dixit 2025-03-17 20:39:09 +05:30 committed by GitHub
parent 27361bd218
commit 8e67230860
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tests/utils/test_import_utils.py
View File

@ -1,23 +1,26 @@
import sys import sys
from transformers.testing_utils import run_test_using_subprocess
from transformers.utils.import_utils import clear_import_cache from transformers.utils.import_utils import clear_import_cache
@run_test_using_subprocess
def test_clear_import_cache(): def test_clear_import_cache():
# Import some transformers modules """Test the clear_import_cache function."""
# Get initial module count # Save initial state
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.")}
assert len(initial_modules) > 0, "No transformers modules loaded before test"
# Verify we have some modules loaded # Execute clear_import_cache() function
assert len(initial_modules) > 0
# Clear cache
clear_import_cache() clear_import_cache()
# Check modules were removed # Verify 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) assert len(remaining_modules) < len(initial_modules), "No modules were removed"
# Verify we can reimport # Import and verify module exists
assert "transformers" in sys.modules from transformers.models.auto import modeling_auto
assert "transformers.models.auto.modeling_auto" in sys.modules
assert modeling_auto.__name__ == "transformers.models.auto.modeling_auto"