Skip to content

Commit f4271ef

Browse files
committed
fix(csv loader):
1 parent feb1bec commit f4271ef

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

agentic_security/probe_data/data.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -265,7 +265,7 @@ def load_local_csv() -> ProbeDataset:
265265
return create_probe_dataset("Local CSV", prompts, {"src": str(csv_files)})
266266

267267

268-
@cache_to_disk()
268+
@cache_to_disk(1)
269269
def load_csv(file: str) -> ProbeDataset:
270270
"""Load prompts from local CSV files."""
271271
prompts = []
@@ -281,7 +281,7 @@ def load_csv(file: str) -> ProbeDataset:
281281
return create_probe_dataset(f"fs://{file}", prompts, {"src": str(file)})
282282

283283

284-
@cache_to_disk()
284+
@cache_to_disk(1)
285285
def load_local_csv_files() -> list[ProbeDataset]:
286286
"""Load prompts from local CSV files and return a list of ProbeDataset objects."""
287287
csv_files = [f for f in os.listdir(".") if f.endswith(".csv")]

0 commit comments

Comments
 (0)