fix(tests):

agentic_security/probe_data/msj_data.py

@@ -22,7 +22,7 @@ class ProbeDataset:
         }
 
 
-@cache_to_disk()
+# @cache_to_disk(n_days_to_cache=1)
 def load_dataset_generic(name, getter=lambda x: x["train"]["prompt"]):
     from datasets import load_dataset
 

tests/conftest.py

@@ -1,8 +1,16 @@
 import os
 
 import pytest
+from cache_to_disk import delete_old_disk_caches
+from agentic_security.logutils import logger
 
 
 def pytest_runtest_setup(item):
     if "slow" in item.keywords and not os.getenv("RUN_SLOW_TESTS"):
         pytest.skip("Skipping slow test")
+
+
+@pytest.fixture(autouse=True, scope="session")
+def setup_delete_old_disk_caches():
+    logger.info("delete_old_disk_caches")
+    delete_old_disk_caches()