File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
agentic_security/probe_data Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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 )
269269def 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 )
285285def 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" )]
You can’t perform that action at this time.
0 commit comments