diff --git a/tests/pgle_test.py b/tests/pgle_test.py index 270f3d8df..7f9ea598d 100644 --- a/tests/pgle_test.py +++ b/tests/pgle_test.py @@ -55,6 +55,7 @@ class PgleTest(jtu.JaxTestCase): def tearDown(self): cc.set_cache_dir(None) + cc.reset_cache() super().tearDown() def testPGLEProfilerGetFDOProfile(self):