Make node-id required for quickstart-pytorch (#2712)

Co-authored-by: jafermarq <[email protected]>
adap · Dec 20, 2023 · 1bf2ac3 · 1bf2ac3
1 parent 41cc357
commit 1bf2ac3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/quickstart-pytorch/client.py b/examples/quickstart-pytorch/client.py
@@ -99,6 +99,7 @@ def apply_transforms(batch):
 parser.add_argument(
     "--node-id",
     choices=[0, 1, 2],
+    required=True,
     type=int,
     help="Partition of the dataset divided into 3 iid partitions created artificially.",
 )