Skip to content

Commit

Permalink
Add log info for flower executor (#2900)
Browse files Browse the repository at this point in the history
* Add log info for flower executor

* Fix format
  • Loading branch information
YuanTingHsieh authored Sep 4, 2024
1 parent d07b3f6 commit 60e722a
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions nvflare/app_opt/flower/executor.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,6 +68,15 @@ def get_applet(self, fl_ctx: FLContext):

def configure(self, config: dict, fl_ctx: FLContext):
self.num_rounds = config.get(Constant.CONF_KEY_NUM_ROUNDS)
self._log_states(fl_ctx)

def get_connector_config(self, fl_ctx: FLContext) -> dict:
return {Constant.CONF_KEY_NUM_ROUNDS: self.num_rounds}

def _log_states(self, fl_ctx: FLContext):
self.log_info(fl_ctx, "States are:")
self.log_info(fl_ctx, f" {self.int_server_grpc_options=}")
self.log_info(fl_ctx, f" {self.per_msg_timeout=}")
self.log_info(fl_ctx, f" {self.tx_timeout=}")
self.log_info(fl_ctx, f" {self.client_shutdown_timeout=}")
self.log_info(fl_ctx, f" {self.extra_env=}")

0 comments on commit 60e722a

Please sign in to comment.