diff --git a/src/py/flwr/client/app.py b/src/py/flwr/client/app.py index fa446afcc1fb..d2d5a79f32f3 100644 --- a/src/py/flwr/client/app.py +++ b/src/py/flwr/client/app.py @@ -298,7 +298,7 @@ def _on_backoff(retry_state: RetryState) -> None: retry_invoker = RetryInvoker( wait_gen_factory=exponential, recoverable_exceptions=connection_error_type, - max_tries=max_retries, + max_tries=max_retries + 1 if max_retries is not None else None, max_time=max_wait_time, on_giveup=lambda retry_state: ( log( diff --git a/src/py/flwr/client/supernode/app.py b/src/py/flwr/client/supernode/app.py index 20e0c44eab14..4ee3544417aa 100644 --- a/src/py/flwr/client/supernode/app.py +++ b/src/py/flwr/client/supernode/app.py @@ -345,8 +345,8 @@ def _parse_args_common(parser: argparse.ArgumentParser) -> None: "--max-retries", type=int, default=None, - help="The maximum number of times the client will try to connect to the" - "server before giving up in case of a connection error. By default," + help="The maximum number of times the client will try to reconnect to the" + "SuperLink before giving up in case of a connection error. By default," "it is set to None, meaning there is no limit to the number of tries.", ) parser.add_argument( @@ -354,7 +354,7 @@ def _parse_args_common(parser: argparse.ArgumentParser) -> None: type=float, default=None, help="The maximum duration before the client stops trying to" - "connect to the server in case of connection error. By default, it" + "connect to the SuperLink in case of connection error. By default, it" "is set to None, meaning there is no limit to the total time.", ) parser.add_argument(