mirror of
https://github.com/msoedov/agentic_security.git
synced 2026-06-28 08:09:57 +02:00
fix(tests):
This commit is contained in:
@@ -22,7 +22,7 @@ class ProbeDataset:
|
||||
}
|
||||
|
||||
|
||||
@cache_to_disk()
|
||||
# @cache_to_disk(n_days_to_cache=1)
|
||||
def load_dataset_generic(name, getter=lambda x: x["train"]["prompt"]):
|
||||
from datasets import load_dataset
|
||||
|
||||
|
||||
Reference in New Issue
Block a user