fix: Fix aiplatform.init bug by replacing experiment_name with experiment

vertex-sdk-bot · copybara-github · commit c60773a7db8c · 2023-06-23T16:41:34.000-07:00
PiperOrigin-RevId: 542988570
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/samples/model-builder/experiment_tracking/autologging_with_manual_run_creation_sample_test.py b/samples/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,
     )

Original file line number	Diff line number	Diff line change
`@@ -27,7 +27,7 @@ def test_autologging_with_auto_run_creation_sample(mock_sdk_init, mock_autolog):`
`27`	`27`	`)`
`28`	`28`
`29`	`29`	`mock_sdk_init.assert_called_with(`
`30`		`- experiment_name=constants.EXPERIMENT_NAME,`
	`30`	`+ experiment=constants.EXPERIMENT_NAME,`
`31`	`31`	`project=constants.PROJECT,`
`32`	`32`	`location=constants.LOCATION,`
`33`	`33`	`experiment_tensorboard=constants.TENSORBOARD_NAME,`
Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@ def test_autologging_with_manual_run_creation_sample(`
`30`	`30`	`)`
`31`	`31`
`32`	`32`	`mock_sdk_init.assert_called_with(`
`33`		`- experiment_name=constants.EXPERIMENT_NAME,`
	`33`	`+ experiment=constants.EXPERIMENT_NAME,`
`34`	`34`	`project=constants.PROJECT,`
`35`	`35`	`location=constants.LOCATION,`
`36`	`36`	`experiment_tensorboard=constants.TENSORBOARD_NAME,`
Original file line number	Diff line number	Diff line change
`@@ -27,7 +27,7 @@ def test_create_experiment_run_sample(mock_sdk_init, mock_start_run):`
`27`	`27`	`)`
`28`	`28`
`29`	`29`	`mock_sdk_init.assert_called_with(`
`30`		`- experiment_name=constants.EXPERIMENT_NAME,`
	`30`	`+ experiment=constants.EXPERIMENT_NAME,`
`31`	`31`	`project=constants.PROJECT,`
`32`	`32`	`location=constants.LOCATION,`
`33`	`33`	`)`
Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@ def test_end_experiment_run_sample(mock_sdk_init, mock_start_run, mock_end_run):`
`26`	`26`	`)`
`27`	`27`
`28`	`28`	`mock_sdk_init.assert_called_with(`
`29`		`- experiment_name=constants.EXPERIMENT_NAME,`
	`29`	`+ experiment=constants.EXPERIMENT_NAME,`
`30`	`30`	`project=constants.PROJECT,`
`31`	`31`	`location=constants.LOCATION,`
`32`	`32`	`)`
Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@ def test_resume_experiment_run_sample(mock_sdk_init, mock_start_run):`
`26`	`26`	`)`
`27`	`27`
`28`	`28`	`mock_sdk_init.assert_called_with(`
`29`		`- experiment_name=constants.EXPERIMENT_NAME,`
	`29`	`+ experiment=constants.EXPERIMENT_NAME,`
`30`	`30`	`project=constants.PROJECT,`
`31`	`31`	`location=constants.LOCATION,`
`32`	`32`	`)`