From cba7fbfba6f06273aa3f95911c92d25720009c28 Mon Sep 17 00:00:00 2001 From: sasha-gitg <44654632+sasha-gitg@users.noreply.github.com> Date: Wed, 5 Oct 2022 11:34:03 -0700 Subject: [PATCH] docs: fix create experiment sample (#1716) --- .../experiment_tracking/create_experiment_sample.py | 2 +- .../experiment_tracking/create_experiment_sample_test.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/samples/model-builder/experiment_tracking/create_experiment_sample.py b/samples/model-builder/experiment_tracking/create_experiment_sample.py index f73954cc95..1b655f1c53 100644 --- 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 index c4c3ae3843..73774c5a15 100644 --- 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,