Apply suggestions from code review

adap · Mar 13, 2024 · b9392aa · b9392aa
1 parent bd04e37
commit b9392aa
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/datasets/flwr_datasets/federated_dataset.py b/datasets/flwr_datasets/federated_dataset.py
@@ -228,16 +228,16 @@ def concatenate_divisions(
 
         Parameters
         ----------
-        division_id: Union[int, str]
+        division_id : Union[int, str]
             The way to access the division (from a List or DatasetDict).
-        split: Optional[str]
+        split : Optional[str]
             Split associated with a partitioner that creates the division. It needs to
             be specified if `partition_division` specifies divisions for more than
             one partitioner.
 
         Returns
         -------
-        concatenated_divisions: Dataset
+        concatenated_divisions : Dataset
             A dataset created as concatenation of the divisions from all partitions.
         """
         if self._partition_division is None: