From 24e146551237c494349b324ee8830154d129860c Mon Sep 17 00:00:00 2001 From: Sasha Sobran Date: Fri, 2 Dec 2022 11:24:44 -0800 Subject: [PATCH] docs: fix get Experiment data frame sample PiperOrigin-RevId: 492516226 --- .../experiment_tracking/get_experiment_data_frame_sample.py | 4 ++-- .../get_experiment_data_frame_sample_test.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample.py b/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample.py index 0a65a82f7b..5b3706cc7c 100644 --- a/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample.py +++ b/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample.py @@ -18,11 +18,11 @@ # [START aiplatform_sdk_get_experiments_data_frame_sample] def get_experiments_data_frame_sample( - experiment_name: str, + experiment: str, project: str, location: str, ): - aiplatform.init(experiment_name=experiment_name, project=project, location=location) + aiplatform.init(experiment=experiment, project=project, location=location) experiments_df = aiplatform.get_experiment_df() diff --git a/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample_test.py b/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample_test.py index 8740f5c2c7..b042de0bfc 100644 --- a/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample_test.py +++ b/samples/model-builder/experiment_tracking/get_experiment_data_frame_sample_test.py @@ -22,7 +22,7 @@ @pytest.mark.usefixtures("mock_sdk_init") def test_get_experiments_data_frame_sample(mock_get_experiment_df, mock_df): df = get_experiment_data_frame_sample.get_experiments_data_frame_sample( - experiment_name=constants.EXPERIMENT_NAME, + experiment=constants.EXPERIMENT_NAME, project=constants.PROJECT, location=constants.LOCATION, )