Remove TBReceiver from tf job api

NVIDIA · Oct 14, 2024 · 49194b0 · 49194b0
1 parent fa135de
commit 49194b0
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/nvflare/app_opt/tf/job_config/base_fed_job.py b/nvflare/app_opt/tf/job_config/base_fed_job.py
@@ -23,7 +23,6 @@
 from nvflare.app_common.widgets.streaming import AnalyticsReceiver
 from nvflare.app_common.widgets.validation_json_generator import ValidationJsonGenerator
 from nvflare.app_opt.tf.job_config.model import TFModel
-from nvflare.app_opt.tracking.tb.tb_receiver import TBAnalyticsReceiver
 from nvflare.job_config.api import FedJob, validate_object_for_job
 
 
@@ -94,13 +93,10 @@ def __init__(
 
         if analytics_receiver:
             validate_object_for_job("analytics_receiver", analytics_receiver, AnalyticsReceiver)
-        else:
-            analytics_receiver = TBAnalyticsReceiver()
-
-        self.to_server(
-            id="receiver",
-            obj=analytics_receiver,
-        )
+            self.to_server(
+                id="receiver",
+                obj=analytics_receiver,
+            )
 
         if initial_model:
             self.comp_ids["persistor_id"] = self.to_server(TFModel(model=initial_model, persistor=model_persistor))