Fix async method error

Signed-off-by: Aleksandr Mokrov <[email protected]>
securefederatedai · Jan 26, 2023 · 33a14fb · 33a14fb
1 parent 954d675
commit 33a14fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openfl/component/director/director.py b/openfl/component/director/director.py
@@ -122,7 +122,7 @@ async def get_experiment_status(
             return None
         return self.experiments_registry[experiment_name].status
 
-    def get_trained_model(self, experiment_name: str, caller: str, model_type: str):
+    async def get_trained_model(self, experiment_name: str, caller: str, model_type: str):
         """Get trained model."""
         if (experiment_name not in self.experiments_registry
                 or caller not in self.experiments_registry[experiment_name].users):