Fix formatting errors

adap · Feb 16, 2024 · cdfedfd · cdfedfd
1 parent d5c22e5
commit cdfedfd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datasets/flwr_datasets/resplitter/divide_resplitter.py b/datasets/flwr_datasets/resplitter/divide_resplitter.py
@@ -91,6 +91,7 @@ def __call__(self, dataset: DatasetDict) -> DatasetDict:
             self._check_duplicate_splits_in_config_and_original_dataset(dataset_splits)
         return self.resplit(dataset)
 
+    # pylint: disable=too-many-branches
     def resplit(self, dataset: DatasetDict) -> DatasetDict:
         """Resplit the dataset according to the configuration."""
         resplit_dataset = {}
@@ -105,8 +106,7 @@ def resplit(self, dataset: DatasetDict) -> DatasetDict:
                         "dataset with more than one split you need to specify the "
                         "`divide_split` but given None instead."
                     )
-                else:
-                    self._divide_split = dataset_splits[0]
+                self._divide_split = dataset_splits[0]
             self._multiple_splits_config = cast(
                 Union[Dict[str, Dict[str, float]], Dict[str, Dict[str, int]]],
                 {self._divide_split: self._single_split_config},