fix: flaky tests in test_utils caused by caching

meetbryce · Oct 11, 2024 · c0066e9 · c0066e9
1 parent 15a8a1e
commit c0066e9
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -8,6 +8,13 @@
 from ossai import utils
 
 
+@pytest.fixture(autouse=True)
+def clear_cache():
+    utils._id_name_cache.clear()
+    yield
+    utils._id_name_cache.clear()
+
+
 @pytest.fixture
 def mock_client():
     with patch("ossai.utils.WebClient") as mock_client: