From 49194b0ad09a3add635b21db53b7bf10aad9a69a Mon Sep 17 00:00:00 2001 From: YuanTingHsieh Date: Sun, 13 Oct 2024 22:44:19 -0700 Subject: [PATCH] Remove TBReceiver from tf job api --- nvflare/app_opt/tf/job_config/base_fed_job.py | 12 ++++-------- 1 file changed, 4 insertions(+), 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 index bf77cd1092..56b6871199 100644 --- 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))