refactor(datasets): Adjust datasets versions

datasets/flwr_datasets/mock_utils_test.py

@@ -334,7 +334,7 @@ def _mock_speach_commands(num_rows: int) -> Dataset:
     features = Features(
         {
             "audio": datasets.Audio(
-                sampling_rate=sampling_rate, mono=True, decode=True
+                sampling_rate=sampling_rate, num_channels=1, decode=True
             ),
             "is_unknown": Value(dtype="bool"),
             "speaker_id": Value(dtype="string"),

datasets/pyproject.toml

@@ -52,7 +52,7 @@ exclude = ["./**/*_test.py"]
 [tool.poetry.dependencies]
 python = "^3.10"
 numpy = ">=1.26.0,<3.0.0"
-datasets = ">=2.14.6 <=3.1.0"
+datasets = ">=3.0.0, <5.0.0"
 pillow = { version = ">=6.2.1", optional = true }
 soundfile = { version = ">=0.12.1", optional = true }
 librosa = { version = ">=0.10.0.post2", optional = true }