docs: fix get Experiment data frame sample

PiperOrigin-RevId: 492516226
googleapis · Dec 2, 2022 · 24e1465 · 24e1465
1 parent 4d3107d
commit 24e1465
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 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
@@ -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
@@ -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,
     )