From cdfedfdb73d0f1c75c61ea44688d47769b447cfe Mon Sep 17 00:00:00 2001 From: Adam Narozniak Date: Fri, 16 Feb 2024 09:43:03 +0100 Subject: [PATCH] Fix formatting errors --- datasets/flwr_datasets/resplitter/divide_resplitter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/datasets/flwr_datasets/resplitter/divide_resplitter.py b/datasets/flwr_datasets/resplitter/divide_resplitter.py index d0d9d8353f83..3b901efd0470 100644 --- 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},