From c766a581826fef6011acb7501ac194f0ea2647b8 Mon Sep 17 00:00:00 2001 From: Sarthak Pati Date: Wed, 14 Feb 2024 18:23:14 -0500 Subject: [PATCH] Updated configuration parsing api for GaNDLF (#920) * updated api * Update runner_gandlf.py Signed-off-by: nammbash --- openfl/federated/task/runner_gandlf.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/openfl/federated/task/runner_gandlf.py b/openfl/federated/task/runner_gandlf.py index 19da03f55cb..1553e51b6ba 100644 --- 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,