Skip to content

Commit

Permalink
Update doc strings
Browse files Browse the repository at this point in the history
  • Loading branch information
sanaAyrml committed Jan 9, 2025
1 parent bf2504d commit 476343e
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions fl4health/clients/basic_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -160,7 +160,7 @@ def get_parameters(self, config: Config) -> NDArrays:
return FullParameterExchanger().push_parameters(self.model, config=config)
else:
if hasattr(self, "early_stopper") and self.early_stopper.patience == 0:
log(INFO, "Loading save best model state before sending model to server.")
log(INFO, "Loading saved best model's state before sending model to server.")
self.early_stopper.load_snapshot(["model"])
assert self.model is not None and self.parameter_exchanger is not None
return self.parameter_exchanger.push_parameters(self.model, config=config)
Expand Down Expand Up @@ -906,10 +906,9 @@ def set_early_stopper(self) -> None:
set self.early_stopper to an instance of EarlyStopper. The EarlyStopper class is defined in
fl4health.early_stopping. Example implementation:
⁠ python
from fl4health.utils.early_stopper import EarlyStopper
self.early_stopper = EarlyStopper(client=self, patience=3, interval_steps=100)
 ⁠
"""
raise NotImplementedError

Expand Down

0 comments on commit 476343e

Please sign in to comment.