From 1c61eaad97dcc9f5b6d12d59d07434b49b9a7aee Mon Sep 17 00:00:00 2001 From: Adam Narozniak Date: Fri, 18 Oct 2024 12:38:28 +0200 Subject: [PATCH] Fix the script based mock downloads --- datasets/flwr_datasets/mock_utils_test.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/datasets/flwr_datasets/mock_utils_test.py b/datasets/flwr_datasets/mock_utils_test.py index 0976166648eb..acfa4b16e4ee 100644 --- 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