docs: fix create experiment sample (#1716)

googleapis · Oct 5, 2022 · cba7fbf · cba7fbf
1 parent f28988e
commit cba7fbf
Show file tree

Hide file tree

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

diff --git a/samples/model-builder/experiment_tracking/create_experiment_sample_test.py b/samples/model-builder/experiment_tracking/create_experiment_sample_test.py
@@ -29,7 +29,7 @@ def test_create_experiment_sample(mock_sdk_init):
     )
 
     mock_sdk_init.assert_called_with(
-        experiment_name=constants.EXPERIMENT_NAME,
+        experiment=constants.EXPERIMENT_NAME,
         experiment_description=constants.DESCRIPTION,
         experiment_tensorboard=constants.TENSORBOARD_NAME,
         project=constants.PROJECT,