Skip to content

Commit

Permalink
Update the partitioners property docs
Browse files Browse the repository at this point in the history
  • Loading branch information
adam-narozniak committed Mar 13, 2024
1 parent 3c9885d commit 07a825e
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion datasets/flwr_datasets/federated_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -163,7 +163,13 @@ def load_full(self, split: str) -> Dataset:

@property
def partitioners(self) -> Dict[str, Partitioner]:
"""Dictionary mapping split to associated partitioners."""
"""Dictionary mapping split to associated partitioners.
The returned partitioners have the splits of the dataset assigned to them.
"""
# This function trigger the dataset download (laizy download) and checks
# the partitioner specification correctness (which can also happen lazily only
# after the dataset download).
if not self._dataset_prepared:
self._prepare_dataset()
if self._dataset is None:
Expand Down

0 comments on commit 07a825e

Please sign in to comment.