From 1bf2ac3580426eeb9cd2f02d8da13935bc568a52 Mon Sep 17 00:00:00 2001
From: Pritam Neog <neog.geek123@gmail.com>
Date: Wed, 20 Dec 2023 21:32:13 +0530
Subject: [PATCH] Make `node-id` required for quickstart-pytorch (#2712)

Co-authored-by: jafermarq <jafermarq@gmail.com>
---
 examples/quickstart-pytorch/client.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/quickstart-pytorch/client.py b/examples/quickstart-pytorch/client.py
index ad57645002f8..1edb42d1ec81 100644
--- 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.",
 )