Skip to content

Commit

Permalink
ruff linting fix
Browse files Browse the repository at this point in the history
  • Loading branch information
FrankJonasmoelle committed Dec 19, 2024
1 parent 6e857bf commit 6c1de24
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions examples/splitlearning_titanic/client/backward.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,10 +17,10 @@


def backward_pass(gradient_path, client_id):
"""Load gradients from in_gradients_path, load the embeddings, and perform a backward pass to update
"""Load gradients from in_gradients_path, load the embeddings, and perform a backward pass to update
the parameters of the client model. Save the updated model to out_model_path.
"""
# load client model with parameters
# load client model with parameters
client_model = load_client_model(client_id)
logger.info(f"Client model loaded from {client_id}")

Expand Down
2 changes: 1 addition & 1 deletion fedn/network/combiner/aggregators/splitlearningagg.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ def __init__(self, update_handler):
def combine_models(self, helper=None, delete_models=True):
"""Concatenates client embeddings in the queue by aggregating them.
After all embeddings are received, the embeddings need to be sorted
After all embeddings are received, the embeddings need to be sorted
(consistently) by client ID.
:param helper: An instance of :class: `fedn.utils.helpers.helpers.HelperBase`, ML framework specific helper, defaults to None
Expand Down
2 changes: 1 addition & 1 deletion fedn/network/controller/control.py
Original file line number Diff line number Diff line change
Expand Up @@ -296,7 +296,7 @@ def splitlearning_session(self, config: RoundConfig) -> None:

if self.get_session_status(config["session_id"]) == "Started":
self.set_session_status(config["session_id"], "Finished")
self._state = ReducerState.idle
self._state = ReducerState.idle

def round(self, session_config: RoundConfig, round_id: str):
"""Execute one global round.
Expand Down

0 comments on commit 6c1de24

Please sign in to comment.