Update datasets/flwr_datasets/partitioner/natural_id_partitioner.py

Co-authored-by: Javier <[email protected]>
adap · May 6, 2024 · 283388d · 283388d
1 parent be1d63f
commit 283388d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/datasets/flwr_datasets/partitioner/natural_id_partitioner.py b/datasets/flwr_datasets/partitioner/natural_id_partitioner.py
@@ -49,7 +49,7 @@ class NaturalIdPartitioner(Partitioner):
     >>> from flwr_datasets import FederatedDataset
     >>> from flwr_datasets.partitioner import NaturalIdPartitioner
     >>>
-    >>> partitioner = NaturalIdPartitioner(partition_by="character_id")
+    >>> partitioner = NaturalIdPartitioner(partition_by="user")
     >>> fds = FederatedDataset(dataset="sentiment140",
     >>>                        partitioners={"train": partitioner})
     >>> partition = fds.load_partition(0)