feat(datasets) Bump up datasets version (#4340)

adap · Oct 18, 2024 · f75a039 · f75a039
1 parent e827ded
commit f75a039
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
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

diff --git a/datasets/pyproject.toml b/datasets/pyproject.toml
@@ -53,7 +53,7 @@ exclude = [
 [tool.poetry.dependencies]
 python = "^3.8"
 numpy = "^1.21.0"
-datasets = ">=2.14.6 <2.20.0"
+datasets = ">=2.14.6 <=3.1.0"
 pillow = { version = ">=6.2.1", optional = true }
 soundfile = { version = ">=0.12.1", optional = true }
 librosa = { version = ">=0.10.0.post2", optional = true }