Update arg name for MLflowReceiver (#2850)

Co-authored-by: Chester Chen <512707+chesterxgchen@users.noreply.github.com>
NVIDIA · Aug 27, 2024 · d26b96c · d26b96c
1 parent 27b7b9e
commit d26b96c
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/examples/advanced/experiment-tracking/mlflow/experiment_tracking.ipynb b/examples/advanced/experiment-tracking/mlflow/experiment_tracking.ipynb
@@ -63,7 +63,7 @@
     "      \"id\": \"mlflow_receiver\",\n",
     "      \"path\": \"nvflare.app_opt.tracking.mlflow.mlflow_receiver.MLflowReceiver\",\n",
     "      \"args\": {\n",
-    "        \"kwargs\": {\"experiment_name\": \"hello-pt-experiment\"},\n",
+    "        \"kw_args\": {\"experiment_name\": \"hello-pt-experiment\"},\n",
     "        \"artifact_location\": \"artifacts\"\n",
     "      }\n",
     "}\n",

diff --git a/examples/hello-world/ml-to-fl/np/np_client_api_job.py b/examples/hello-world/ml-to-fl/np/np_client_api_job.py
@@ -54,7 +54,7 @@ def main():
     if script == "src/train_metrics.py":
         receiver = MLflowReceiver(
             tracking_uri="file:///tmp/nvflare/jobs/workdir/server/simulate_job/mlruns",
-            kwargs={
+            kw_args={
                 "experiment_name": "nvflare-fedavg-np-experiment",
                 "run_name": "nvflare-fedavg-np-with-mlflow",
                 "experiment_tags": {"mlflow.note.content": "## **NVFlare FedAvg Numpy experiment with MLflow**"},

diff --git a/integration/monai/examples/mednist/job_templates/fedavg_mednist/config_fed_server.conf b/integration/monai/examples/mednist/job_templates/fedavg_mednist/config_fed_server.conf
@@ -75,7 +75,7 @@
           args {
             # tracking_uri = "http://0.0.0.0:5000"
             tracking_uri = ""
-            kwargs {
+            kw_args {
               experiment_name =  "nvflare-fedavg-mednist-experiment"
               run_name = "nvflare-fedavg-mednist-with-mlflow"
               experiment_tags {

diff --git a/...ct_segmentation_local/jobs/spleen_ct_segmentation_local/app/config/config_fed_server.conf b/...ct_segmentation_local/jobs/spleen_ct_segmentation_local/app/config/config_fed_server.conf
@@ -48,7 +48,7 @@
       "path": "nvflare.app_opt.tracking.mlflow.mlflow_receiver.MLflowReceiver",
       "args": {
         "tracking_uri": "http://127.0.0.1:5000",
-        "kwargs": {
+        "kw_args": {
           "experiment_name": "monai-spleen-experiment",
           "run_name": "monai-spleen-with-mlflow",
           "experiment_tags": {

diff --git a/job_templates/sag_pt_in_proc/config_fed_server.conf b/job_templates/sag_pt_in_proc/config_fed_server.conf
@@ -108,7 +108,7 @@
           args {
             # tracking_uri = "http://0.0.0.0:5000"
             tracking_uri =  "file:///{WORKSPACE}/{JOB_ID}/mlruns"
-            kwargs {
+            kw_args {
               experiment_name =  "nvflare-sag-pt-experiment"
               run_name = "nvflare-sag-pt-with-mlflow"
               experiment_tags {

diff --git a/job_templates/sag_pt_mlflow/config_fed_server.conf b/job_templates/sag_pt_mlflow/config_fed_server.conf
@@ -101,7 +101,7 @@
       path = "nvflare.app_opt.tracking.mlflow.mlflow_receiver.MLflowReceiver"
       args {
         "tracking_uri": "http://0.0.0.0:5000"
-        "kwargs": {
+        "kw_args": {
           "experiment_name": "nvflare-sag-pt-experiment"
           "run_name": "nvflare-sag-pt-with-mlflow"
           "experiment_tags": {

diff --git a/tests/integration_test/data/jobs/pt_client_api_in_process/app/config/config_fed_server.conf b/tests/integration_test/data/jobs/pt_client_api_in_process/app/config/config_fed_server.conf
@@ -63,7 +63,7 @@
       path = "nvflare.app_opt.tracking.mlflow.mlflow_receiver.MLflowReceiver"
       args {
         tracking_uri = ""
-        kwargs {
+        kw_args {
           experiment_name = "nvflare-sag-pt-experiment"
           run_name = "nvflare-sag-pt-with-mlflow"
           experiment_tags {