fix: Fix aiplatform.init bug by replacing experiment_name with experi…

…ment PiperOrigin-RevId: 542988570
googleapis · Jun 23, 2023 · c60773a · c60773a
1 parent c6cdd10
commit c60773a
Show file tree

Hide file tree

Showing 11 changed files with 11 additions and 11 deletions.
diff --git a/samples/model-builder/experiment_tracking/autologging_with_auto_run_creation_sample.py b/samples/model-builder/experiment_tracking/autologging_with_auto_run_creation_sample.py
@@ -25,7 +25,7 @@ def autologging_with_auto_run_creation_sample(
     location: str,
 ):
     aiplatform.init(
-        experiment_name=experiment_name,
+        experiment=experiment_name,
         project=project,
         location=location,
         experiment_tensorboard=experiment_tensorboard,

diff --git a/samples/model-builder/experiment_tracking/autologging_with_auto_run_creation_sample_test.py b/samples/model-builder/experiment_tracking/autologging_with_auto_run_creation_sample_test.py
@@ -27,7 +27,7 @@ def test_autologging_with_auto_run_creation_sample(mock_sdk_init, mock_autolog):
     )
 
     mock_sdk_init.assert_called_with(
-        experiment_name=constants.EXPERIMENT_NAME,
+        experiment=constants.EXPERIMENT_NAME,
         project=constants.PROJECT,
         location=constants.LOCATION,
         experiment_tensorboard=constants.TENSORBOARD_NAME,

diff --git a/samples/model-builder/experiment_tracking/autologging_with_manual_run_creation_sample.py b/samples/model-builder/experiment_tracking/autologging_with_manual_run_creation_sample.py
@@ -26,7 +26,7 @@ def autologging_with_manual_run_creation_sample(
     location: str,
 ):
     aiplatform.init(
-        experiment_name=experiment_name,
+        experiment=experiment_name,
         project=project,
         location=location,
         experiment_tensorboard=experiment_tensorboard,

diff --git a/...les/model-builder/experiment_tracking/autologging_with_manual_run_creation_sample_test.py b/...les/model-builder/experiment_tracking/autologging_with_manual_run_creation_sample_test.py
@@ -30,7 +30,7 @@ def test_autologging_with_manual_run_creation_sample(
     )
 
     mock_sdk_init.assert_called_with(
-        experiment_name=constants.EXPERIMENT_NAME,
+        experiment=constants.EXPERIMENT_NAME,
         project=constants.PROJECT,
         location=constants.LOCATION,
         experiment_tensorboard=constants.TENSORBOARD_NAME,

diff --git a/samples/model-builder/experiment_tracking/create_experiment_run_sample.py b/samples/model-builder/experiment_tracking/create_experiment_run_sample.py
@@ -25,7 +25,7 @@ def create_experiment_run_sample(
     project: str,
     location: str,
 ):
-    aiplatform.init(experiment_name=experiment_name, project=project, location=location)
+    aiplatform.init(experiment=experiment_name, project=project, location=location)
 
     aiplatform.start_run(run=run_name, tensorboard=experiment_run_tensorboard)
 

diff --git a/samples/model-builder/experiment_tracking/create_experiment_run_sample_test.py b/samples/model-builder/experiment_tracking/create_experiment_run_sample_test.py
@@ -27,7 +27,7 @@ def test_create_experiment_run_sample(mock_sdk_init, mock_start_run):
     )
 
     mock_sdk_init.assert_called_with(
-        experiment_name=constants.EXPERIMENT_NAME,
+        experiment=constants.EXPERIMENT_NAME,
         project=constants.PROJECT,
         location=constants.LOCATION,
     )

diff --git a/samples/model-builder/experiment_tracking/end_experiment_run_sample.py b/samples/model-builder/experiment_tracking/end_experiment_run_sample.py
@@ -23,7 +23,7 @@ def end_experiment_run_sample(
     project: str,
     location: str,
 ):
-    aiplatform.init(experiment_name=experiment_name, project=project, location=location)
+    aiplatform.init(experiment=experiment_name, project=project, location=location)
 
     aiplatform.start_run(run=run_name, resume=True)
 

diff --git a/samples/model-builder/experiment_tracking/end_experiment_run_sample_test.py b/samples/model-builder/experiment_tracking/end_experiment_run_sample_test.py
@@ -26,7 +26,7 @@ def test_end_experiment_run_sample(mock_sdk_init, mock_start_run, mock_end_run):
     )
 
     mock_sdk_init.assert_called_with(
-        experiment_name=constants.EXPERIMENT_NAME,
+        experiment=constants.EXPERIMENT_NAME,
         project=constants.PROJECT,
         location=constants.LOCATION,
     )

diff --git a/samples/model-builder/experiment_tracking/log_time_series_metrics_sample.py b/samples/model-builder/experiment_tracking/log_time_series_metrics_sample.py
@@ -28,7 +28,7 @@ def log_time_series_metrics_sample(
     project: str,
     location: str,
 ):
-    aiplatform.init(experiment_name=experiment_name, project=project, location=location)
+    aiplatform.init(experiment=experiment_name, project=project, location=location)
 
     aiplatform.start_run(run_name=run_name, resume=True)
 

diff --git a/samples/model-builder/experiment_tracking/resume_experiment_run_sample.py b/samples/model-builder/experiment_tracking/resume_experiment_run_sample.py
@@ -23,7 +23,7 @@ def resume_experiment_run_sample(
     project: str,
     location: str,
 ):
-    aiplatform.init(experiment_name=experiment_name, project=project, location=location)
+    aiplatform.init(experiment=experiment_name, project=project, location=location)
 
     aiplatform.start_run(run=run_name, resume=True)
 

diff --git a/samples/model-builder/experiment_tracking/resume_experiment_run_sample_test.py b/samples/model-builder/experiment_tracking/resume_experiment_run_sample_test.py
@@ -26,7 +26,7 @@ def test_resume_experiment_run_sample(mock_sdk_init, mock_start_run):
     )
 
     mock_sdk_init.assert_called_with(
-        experiment_name=constants.EXPERIMENT_NAME,
+        experiment=constants.EXPERIMENT_NAME,
         project=constants.PROJECT,
         location=constants.LOCATION,
     )