CalvinBackup/agentic_security
1
0
Fork 0
mirror of https://github.com/msoedov/agentic_security.git synced 2026-06-23 21:59:57 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix(csv loader):

Browse Source
This commit is contained in:
Alexander Myasoedov
2025-03-13 18:32:22 +02:00
parent feb1becb3e
commit f4271ef2a1
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
agentic_security/probe_data/data.py
+2 -2
View File
@@ -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")]