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}")