From e7ca53c198960d5eaed309b66948f064964bd7cf Mon Sep 17 00:00:00 2001 From: Adam Narozniak Date: Mon, 20 May 2024 15:08:52 +0200 Subject: [PATCH] Rename _resplitter to _preprocessor --- datasets/flwr_datasets/federated_dataset.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/datasets/flwr_datasets/federated_dataset.py b/datasets/flwr_datasets/federated_dataset.py index aefa3ccdf9b8..7189627cefa9 100644 --- a/datasets/flwr_datasets/federated_dataset.py +++ b/datasets/flwr_datasets/federated_dataset.py @@ -89,7 +89,7 @@ def __init__( _check_if_dataset_tested(dataset) self._dataset_name: str = dataset self._subset: Optional[str] = subset - self._resplitter: Optional[Preprocessor] = _instantiate_resplitter_if_needed( + self._preprocessor: Optional[Preprocessor] = _instantiate_resplitter_if_needed( preprocessor ) self._partitioners: Dict[str, Partitioner] = _instantiate_partitioners( @@ -244,8 +244,8 @@ def _prepare_dataset(self) -> None: # Note it shuffles all the splits. The self._dataset is DatasetDict # so e.g. {"train": train_data, "test": test_data}. All splits get shuffled. self._dataset = self._dataset.shuffle(seed=self._seed) - if self._resplitter: - self._dataset = self._resplitter(self._dataset) + if self._preprocessor: + self._dataset = self._preprocessor(self._dataset) self._dataset_prepared = True def _check_if_no_split_keyword_possible(self) -> None: