From 4449409b080e26e0eb831c45c61383aa0040a6f0 Mon Sep 17 00:00:00 2001
From: Holger Roth <hroth@nvidia.com>
Date: Thu, 11 Apr 2024 15:16:12 -0400
Subject: [PATCH] run training with in process executor

---
 .../hello-world/python_jobs/client-api-pt/client_api_pt_job.py  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/hello-world/python_jobs/client-api-pt/client_api_pt_job.py b/examples/hello-world/python_jobs/client-api-pt/client_api_pt_job.py
index 5bfa284a13..68486026ea 100644
--- a/examples/hello-world/python_jobs/client-api-pt/client_api_pt_job.py
+++ b/examples/hello-world/python_jobs/client-api-pt/client_api_pt_job.py
@@ -170,7 +170,7 @@ def _create_server_app(self):
     for i in range(n_clients):
         executor = PTInProcessClientAPIExecutor(
             task_fn_path="cifar10_fl.main",
-            #task_fn_args={"batch_size": 1}
+            task_fn_args={}
         )
         client_app = ExecutorApp(executors=[executor])
         job.to(client_app, f"site-{i}")