diff --git a/examples/monai-2D-mednist/client/train.py b/examples/monai-2D-mednist/client/train.py index 90d67add0..2ee922865 100644 --- a/examples/monai-2D-mednist/client/train.py +++ b/examples/monai-2D-mednist/client/train.py @@ -62,7 +62,7 @@ def train(in_model_path, out_model_path, data_path=None, client_settings_path=No batch_size = client_settings["batch_size"] max_epochs = client_settings["local_epochs"] num_workers = client_settings["num_workers"] - split_index = os.environ.get("FEDN_DATA_SPLIT_INDEX") + split_index = os.environ.get("FEDN_DATA_SPLIT_INDEX") lr = client_settings["lr"] if data_path is None: diff --git a/examples/monai-2D-mednist/client/validate.py b/examples/monai-2D-mednist/client/validate.py index 2f4d3d353..ff4eb9263 100644 --- a/examples/monai-2D-mednist/client/validate.py +++ b/examples/monai-2D-mednist/client/validate.py @@ -45,7 +45,7 @@ def validate(in_model_path, out_json_path, data_path=None, client_settings_path= num_workers = client_settings["num_workers"] batch_size = client_settings["batch_size"] - split_index = os.environ.get("FEDN_DATA_SPLIT_INDEX") + split_index = os.environ.get("FEDN_DATA_SPLIT_INDEX") if data_path is None: data_path = os.environ.get("FEDN_DATA_PATH")