From c5c5ae2e4bf988f89a17ff5b569f956eacb30e4f Mon Sep 17 00:00:00 2001 From: Alexander Myasoedov Date: Mon, 19 May 2025 12:29:28 +0300 Subject: [PATCH] fix(makedir): --- agentic_security/probe_data/data.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/agentic_security/probe_data/data.py b/agentic_security/probe_data/data.py index 378e583..416377c 100644 --- a/agentic_security/probe_data/data.py +++ b/agentic_security/probe_data/data.py @@ -248,6 +248,7 @@ def load_jailbreak_v28k() -> ProbeDataset: @cache_to_disk() def load_local_csv() -> ProbeDataset: """Load prompts from local CSV files.""" + os.makedirs("./datasets", exist_ok=True) csv_files = [f for f in os.listdir("./datasets") if f.endswith(".csv")] logger.info(f"Found {len(csv_files)} CSV files: {csv_files}") @@ -337,7 +338,9 @@ class StenographyTransformer: continue # Convert async generator (if any) to a list - prompts = list(dataset.prompts) if hasattr(dataset.prompts, '__iter__') else [] + prompts = ( + list(dataset.prompts) if hasattr(dataset.prompts, "__iter__") else [] + ) size = len(prompts) if size == 0: