From d6f1f587db1f0db59711c9ead568face8fccaa5e Mon Sep 17 00:00:00 2001
From: Holger Roth <hroth@nvidia.com>
Date: Fri, 26 Apr 2024 15:57:10 -0400
Subject: [PATCH] call init later

---
 .../monai/examples/mednist/code/monai_mednist_train.py      | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/integration/monai/examples/mednist/code/monai_mednist_train.py b/integration/monai/examples/mednist/code/monai_mednist_train.py
index bb8dfb41ea..45033c2aa2 100644
--- a/integration/monai/examples/mednist/code/monai_mednist_train.py
+++ b/integration/monai/examples/mednist/code/monai_mednist_train.py
@@ -52,9 +52,6 @@
 
 print_config()
 
-# (2) initializes NVFlare client API
-flare.init()
-
 # Setup data directory
 directory = os.environ.get("MONAI_DATA_DIRECTORY")
 root_dir = tempfile.mkdtemp() if directory is None else directory
@@ -97,6 +94,9 @@
     train_handlers=StatsHandler(),
 )
 
+# (2) initializes NVFlare client API
+flare.init()
+
 # (optional) calculate total steps
 steps = max_epochs * len(train_loader)
 # Run the training