Updated configuration parsing api for GaNDLF (securefederatedai#920)

* updated api * Update runner_gandlf.py Signed-off-by: nammbash <[email protected]>
nammbash · Feb 29, 2024 · c766a58 · c766a58
1 parent a75dc64
commit c766a58
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/openfl/federated/task/runner_gandlf.py b/openfl/federated/task/runner_gandlf.py
@@ -19,7 +19,7 @@
 from GANDLF.compute.generic import create_pytorch_objects
 from GANDLF.compute.training_loop import train_network
 from GANDLF.compute.forward_pass import validate_network
-from GANDLF.parseConfig import parseConfig
+from GANDLF.config_manager import ConfigManager
 
 
 class GaNDLFTaskRunner(TaskRunner):
@@ -49,9 +49,9 @@ def __init__(
             gandlf_config = yaml.safe_load(open(gandlf_config, "r"))
 
         try:
-            gandlf_config = parseConfig(gandlf_config)
+            gandlf_config = ConfigManager(gandlf_config)
         except Exception:
-            self.logger.info("WARNING: GANDLF.parseConfig did not work as expected.")
+            self.logger.info("WARNING: GANDLF.config_manager.ConfigManager did not work.")
 
         (
             model,