Fix the script based mock downloads

adap · Oct 18, 2024 · 1c61eaa · 1c61eaa
1 parent a03ff26
commit 1c61eaa
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/datasets/flwr_datasets/mock_utils_test.py b/datasets/flwr_datasets/mock_utils_test.py
@@ -409,7 +409,11 @@ def _load_mocked_dataset_by_partial_download(
         The dataset with the requested samples.
     """
     dataset = datasets.load_dataset(
-        dataset_name, name=subset_name, split=split_name, streaming=True
+        dataset_name,
+        name=subset_name,
+        split=split_name,
+        streaming=True,
+        trust_remote_code=True,
     )
     dataset_list = []
     # It's a list of dict such that each dict represent a single sample of the dataset