docs: fix create experiment sample (#1716)

sasha-gitg · web-flow · commit cba7fbfba6f0 · 2022-10-05T11:34:03.000-07:00
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,

Original file line number	Diff line number	Diff line change
`@@ -29,7 +29,7 @@ def test_create_experiment_sample(mock_sdk_init):`
`29`	`29`	`)`
`30`	`30`
`31`	`31`	`mock_sdk_init.assert_called_with(`
`32`		`- experiment_name=constants.EXPERIMENT_NAME,`
	`32`	`+ experiment=constants.EXPERIMENT_NAME,`
`33`	`33`	`experiment_description=constants.DESCRIPTION,`
`34`	`34`	`experiment_tensorboard=constants.TENSORBOARD_NAME,`
`35`	`35`	`project=constants.PROJECT,`