added trust_remote_code=True for get_example_dataset.py

angelolab · Jun 19, 2024 · 5eb7e3b · 5eb7e3b
1 parent 39d09fc
commit 5eb7e3b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/scripts/get_example_dataset.py b/.github/scripts/get_example_dataset.py
@@ -7,7 +7,7 @@
 
 DATASET_PATH = "angelolab/ark_example"
 
-valid_configs = datasets.get_dataset_config_names(DATASET_PATH)
+valid_configs = datasets.get_dataset_config_names(DATASET_PATH, trust_remote_code=True)
 
 def load_dataset(cache_dir: Path, name: str):
     _ = datasets.load_dataset(