minort tweaks to make client.py compatible w/ Flower and Flower Next

adap · Apr 2, 2024 · 0c8e0cb · 0c8e0cb
1 parent fb65b67
commit 0c8e0cb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 15 deletions.
diff --git a/examples/quickstart-pytorch/README.md b/examples/quickstart-pytorch/README.md
@@ -87,7 +87,7 @@ flower-superlink --insecure
 
 ### 2. Start the long-running Flower clients (SuperNodes)
 
-Start 2 long-running Flower clients in 2 separate terminal windows, using:
+Start 2 Flower `SuperNodes` in 2 separate terminal windows, using:
 
 ```bash
 flower-client-app client:app --insecure
@@ -100,15 +100,3 @@ With both the long-running server (SuperLink) and two clients (SuperNode) up and
 ```bash
 flower-server-app server:app --insecure
 ```
-
-Or, to try the workflow example, run:
-
-```bash
-flower-server-app server_workflow:app --insecure
-```
-
-Or, to try the custom server function example, run:
-
-```bash
-flower-server-app server_custom:app --insecure
-```
diff --git a/examples/quickstart-pytorch/client.py b/examples/quickstart-pytorch/client.py
@@ -99,11 +99,11 @@ def apply_transforms(batch):
 parser.add_argument(
     "--partition-id",
     choices=[0, 1, 2],
-    required=True,
+    default=0,
     type=int,
     help="Partition of the dataset divided into 3 iid partitions created artificially.",
 )
-partition_id = parser.parse_args().partition_id
+partition_id = parser.parse_known_args()[0].partition_id
 
 # Load model and data (simple CNN, CIFAR-10)
 net = Net().to(DEVICE)