added function to clean up after tests

ur-whitelab · Feb 28, 2024 · ac0cdb5 · ac0cdb5
1 parent a8a29d7
commit ac0cdb5
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -0,0 +1,10 @@
+from mdagent.utils.clear_mem import ClearMemory
+
+
+def cleanup_tests():
+    clear_mem = ClearMemory()
+    clear_mem.clear_ckpts_root()
+
+
+def pytest_sessionfinish(session, exitstatus):
+    cleanup_tests()