Skip to content

Commit

Permalink
Fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
charlesbvll committed Jul 7, 2024
1 parent fc732f3 commit aa4a617
Show file tree
Hide file tree
Showing 3 changed files with 1 addition and 5 deletions.
4 changes: 1 addition & 3 deletions src/py/flwr/client/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,6 @@
from flwr.client.typing import ClientFnExt
from flwr.common import GRPC_MAX_MESSAGE_LENGTH, EventType, Message, event
from flwr.common.address import parse_address
from flwr.common.config import get_fused_config
from flwr.common.constant import (
MISSING_EXTRA_REST,
TRANSPORT_TYPE_GRPC_ADAPTER,
Expand Down Expand Up @@ -381,14 +380,13 @@ def _on_backoff(retry_state: RetryState) -> None:
run_info[run_id] = get_run(run_id)
# If get_run is None, i.e., in grpc-bidi mode
else:
run_info[run_id] = Run(run_id, "", "", {})
run_info[run_id] = Run(run_id, "", "")

# Register context for this run
node_state.register_context(run_id=run_id)

# Retrieve context for this run
context = node_state.retrieve_context(run_id=run_id)
context.run_config = get_fused_config(run_info[run_id], flwr_dir)

# Create an error reply message that will never be used to prevent
# the used-before-assignment linting error
Expand Down
1 change: 0 additions & 1 deletion src/py/flwr/client/grpc_rere_client/connection.py
Original file line number Diff line number Diff line change
Expand Up @@ -280,7 +280,6 @@ def get_run(run_id: int) -> Run:
run_id,
get_run_response.run.fab_id,
get_run_response.run.fab_version,
dict(get_run_response.run.override_config.items()),
)

try:
Expand Down
1 change: 0 additions & 1 deletion src/py/flwr/client/rest_client/connection.py
Original file line number Diff line number Diff line change
Expand Up @@ -358,7 +358,6 @@ def get_run(run_id: int) -> Run:
run_id,
res.run.fab_id,
res.run.fab_version,
dict(res.run.override_config.items()),
)

try:
Expand Down

0 comments on commit aa4a617

Please sign in to comment.