From 0c57cd2dcb2b191b5970a19f21efe343aba97bbd Mon Sep 17 00:00:00 2001 From: Aleksandr Mokrov Date: Mon, 12 Dec 2022 17:51:13 +0100 Subject: [PATCH] Envoy can reconnect to current experiment after restarting for the duration of the experiment. Signed-off-by: Aleksandr Mokrov --- openfl/component/director/director.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/openfl/component/director/director.py b/openfl/component/director/director.py index 52b36e0b3d..0a58654337 100644 --- a/openfl/component/director/director.py +++ b/openfl/component/director/director.py @@ -132,7 +132,7 @@ async def wait_experiment(self, envoy_name: str) -> str: # Experiment already set, but the envoy hasn't received experiment # name (e.g. was disconnected) experiment = self.experiments_registry[experiment_name] - if experiment.aggregator.round_number == 0: + if experiment.aggregator.round_number < experiment.aggregator.rounds_to_train: return experiment_name self.col_exp[envoy_name] = None