diff --git a/datasets/flwr_datasets/federated_dataset_test.py b/datasets/flwr_datasets/federated_dataset_test.py
index 0500fb5b821c..66071af8e348 100644
--- 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),
         ],
     )