added config parsing check

securefederatedai · Jan 16, 2024 · c70ef8b · c70ef8b
1 parent f1657ab
commit c70ef8b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/openfl/federated/task/runner_gandlf.py b/openfl/federated/task/runner_gandlf.py
@@ -45,6 +45,13 @@ def __init__(
         if isinstance(gandlf_config, str) and os.path.exists(gandlf_config):
             gandlf_config = yaml.safe_load(open(gandlf_config, "r"))
 
+        # if we are here, the assumption is that GaNDLF is present in the environment
+        try:
+            from GANDLF.parseConfig import parseConfig as gandlf_config_parser
+            gandlf_config = gandlf_config_parser(gandlf_config)
+        except Exception:
+            self.logger.info("WARNING: GaNDLF doesn't seem to have been correctly installed")
+
         (
             model,
             optimizer,