From ac0cdb5749ff28c68dc220e2955cb5d4b284271b Mon Sep 17 00:00:00 2001
From: Sam Cox <swrig30@ur.rochester.edu>
Date: Wed, 28 Feb 2024 15:33:38 -0800
Subject: [PATCH] added function to clean up after tests

---
 tests/conftest.py | 10 ++++++++++
 1 file changed, 10 insertions(+)
 create mode 100644 tests/conftest.py

diff --git a/tests/conftest.py b/tests/conftest.py
new file mode 100644
index 00000000..78f5522c
--- /dev/null
+++ 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()