Expose utils module

adap · Mar 5, 2024 · 0ed7585 · 0ed7585
1 parent d08b44b
commit 0ed7585
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/datasets/flwr_datasets/federated_dataset_test.py b/datasets/flwr_datasets/federated_dataset_test.py
@@ -70,15 +70,10 @@ def test_load_partition_size(self, _: str, train_num_partitions: int) -> None:
     @parameterized.expand(  # type: ignore
         [
             ((0.2, 0.8), 2),
-            ((0.2, 0.8), None),
-            ([0.2, 0.8], 2),
             ({"train": 0.2, "test": 0.8}, 2),
             # Not full dataset
             ([0.2, 0.1], 2),
-            ((0.2, 0.1), None),
             ({"train": 0.2, "test": 0.1}, 2),
-            ({"train": 0.2, "test": 0.1}, None),
-            (None, None),
             (None, None),
         ],
     )