fix(tests):

This commit is contained in:
Alexander Myasoedov
2025-04-02 13:31:36 +03:00
parent a8e80e85e1
commit f2b95a0040
3 changed files with 44 additions and 200 deletions
+1 -1
View File
@@ -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