some redesign

examples/hello-world/python_jobs/pt/client_api_pt.py

@@ -22,20 +22,23 @@
     num_rounds = 2
     train_script = "code/cifar10_fl.py"
 
-    job = FedJob(name="cifar10_fedavg", init_model=Net(), external_scripts=[train_script])  # TODO: use load/save model in FedAvg
+    job = FedJob(name="cifar10_fedavg") #, init_model=Net(), external_scripts=[train_script])  # TODO: use load/save model in FedAvg
 
     controller = FedAvg(
         min_clients=n_clients,
         num_rounds=num_rounds,
     )
     job.to(controller, "server")
 
+    # Define the initial server model
+    job.to(Net(), "server")  # TODO: default to PTFileModelPersistor -> persistor
+
     for i in range(n_clients):
         executor = ScriptExecutor(
             task_script_path=train_script,
             task_script_args=""  # f"--batch_size 32 --data_path /tmp/data/site-{i}"
         )
-        job.to(executor, f"site-{i}", gpu=0)
+        job.to(executor, f"site-{i}", gpu=0)  # TODO: get script path from executor
 
     job.export_job("/tmp/nvflare/jobs/job_config")
     job.simulator_run("/tmp/nvflare/jobs/workdir")