diff --git a/tests/test_utils.py b/tests/test_utils.py index 187f814..0d88abe 100644 --- 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: