formatting changes

NVIDIA · Feb 2, 2025 · 3e8211e · 3e8211e
1 parent c743b18
commit 3e8211e
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 9 deletions.
diff --git a/...stem/chapter-3_federated_computing_platform/03.2_deployment_simulation/code/src/fedavg.py b/...stem/chapter-3_federated_computing_platform/03.2_deployment_simulation/code/src/fedavg.py
@@ -36,7 +36,15 @@ class FedAvg(BaseFedAvg):
         initial_model (nn.Module, optional): initial PyTorch model
     """
 
-    def __init__(self, *args, stop_cond: str, num_rounds: int, save_filename: str = "FL_global_model.pt", initial_model=None, **kwargs):
+    def __init__(
+        self,
+        *args,
+        stop_cond: str,
+        num_rounds: int,
+        save_filename: str = "FL_global_model.pt",
+        initial_model=None,
+        **kwargs,
+    ):
         super().__init__(*args, **kwargs)
 
         self.stop_cond = stop_cond

diff --git a/...chapter-3_federated_computing_platform/03.4_system_monitoring/jobs/setup-1/code/fl_job.py b/...chapter-3_federated_computing_platform/03.4_system_monitoring/jobs/setup-1/code/fl_job.py
@@ -15,13 +15,11 @@
 import argparse
 import os
 
-
-from src.server import FedAvg
 from src.network import SimpleNetwork
-
-from nvflare.job_config.api import FedJob
+from src.server import FedAvg
 
 from nvflare.fuel_opt.statsd.statsd_reporter import StatsDReporter
+from nvflare.job_config.api import FedJob
 from nvflare.job_config.script_runner import ScriptRunner
 from nvflare.metrics.job_metrics_collector import JobMetricsCollector
 
@@ -40,7 +38,6 @@ def main(job_configs_dir):
     train_script = "src/client.py"
     job_name = "fedavg"
 
-
     job = FedJob(name=job_name, min_clients=num_clients)
 
     controller = FedAvg(

diff --git a/...chapter-3_federated_computing_platform/03.4_system_monitoring/jobs/setup-2/code/fl_job.py b/...chapter-3_federated_computing_platform/03.4_system_monitoring/jobs/setup-2/code/fl_job.py
@@ -15,13 +15,12 @@
 import argparse
 import os
 
-from src.server import FedAvg
 from src.network import SimpleNetwork
-
-from nvflare.job_config.api import FedJob
+from src.server import FedAvg
 
 from nvflare.app_common.widgets.convert_to_fed_event import ConvertToFedEvent
 from nvflare.fuel_opt.statsd.statsd_reporter import StatsDReporter
+from nvflare.job_config.api import FedJob
 from nvflare.job_config.script_runner import ScriptRunner
 from nvflare.metrics.job_metrics_collector import JobMetricsCollector
 from nvflare.metrics.metrics_keys import METRICS_EVENT_TYPE