docs: fix get Experiment data frame sample

sasha-gitg · copybara-github · commit 24e146551237 · 2022-12-02T11:26:22.000-08:00
PiperOrigin-RevId: 492516226
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,
     )

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@`
`22`	`22`	`@pytest.mark.usefixtures("mock_sdk_init")`
`23`	`23`	`def test_get_experiments_data_frame_sample(mock_get_experiment_df, mock_df):`
`24`	`24`	`df = get_experiment_data_frame_sample.get_experiments_data_frame_sample(`
`25`		`- experiment_name=constants.EXPERIMENT_NAME,`
	`25`	`+ experiment=constants.EXPERIMENT_NAME,`
`26`	`26`	`project=constants.PROJECT,`
`27`	`27`	`location=constants.LOCATION,`
`28`	`28`	`)`