Skip to content

Commit

Permalink
fix docstrings example
Browse files Browse the repository at this point in the history
  • Loading branch information
jafermarq committed Dec 19, 2024
1 parent 333ae86 commit 382fbd6
Showing 1 changed file with 6 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -62,18 +62,21 @@ class VerticalEvenPartitioner(Partitioner):
Examples
--------
>>> from flwr_datasets import FederatedDataset
>>> from flwr_datasets.partitioner import VerticalEvenPartitioner
>>>
>>> partitioner = VerticalEvenPartitioner(
... num_partitions=3,
... active_party_column="income",
... active_party_column_mode="add_to_last",
... active_party_columns="income",
... active_party_columns_mode="add_to_last",
... shuffle=True,
... seed=42
... )
>>> fds = FederatedDataset(
... dataset="scikit-learn/adult-census-income",
... partitioners={"train": partitioner}
... )
>>> partitions = [fds.load_partition(i) for i in range(partitioner.num_partitions)]
>>> partitions = [fds.load_partition(i) for i in range(fds.partitioners["train"].num_partitions)]
>>> print([partition.column_names for partition in partitions])
"""

Expand Down

0 comments on commit 382fbd6

Please sign in to comment.