From f4271ef2a1a1de955927ddd3698e7f47c0b7dd85 Mon Sep 17 00:00:00 2001 From: Alexander Myasoedov Date: Thu, 13 Mar 2025 18:32:22 +0200 Subject: [PATCH] fix(csv loader): --- agentic_security/probe_data/data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/agentic_security/probe_data/data.py b/agentic_security/probe_data/data.py index 4844cb1..96ebfd4 100644 --- a/agentic_security/probe_data/data.py +++ b/agentic_security/probe_data/data.py @@ -265,7 +265,7 @@ def load_local_csv() -> ProbeDataset: return create_probe_dataset("Local CSV", prompts, {"src": str(csv_files)}) -@cache_to_disk() +@cache_to_disk(1) def load_csv(file: str) -> ProbeDataset: """Load prompts from local CSV files.""" prompts = [] @@ -281,7 +281,7 @@ def load_csv(file: str) -> ProbeDataset: return create_probe_dataset(f"fs://{file}", prompts, {"src": str(file)}) -@cache_to_disk() +@cache_to_disk(1) def load_local_csv_files() -> list[ProbeDataset]: """Load prompts from local CSV files and return a list of ProbeDataset objects.""" csv_files = [f for f in os.listdir(".") if f.endswith(".csv")]