fix: issues/192254729 (googleapis#914)

andrewferlitsch · web-flow · commit 3ec620c64bd6 · 2021-12-17T00:34:14.000Z
Thank you for opening a Pull Request! Before submitting your PR, there are a few things you can do to make sure it goes smoothly: - [x] Make sure to open an issue as a [bug/issue](https://github.com/googleapis/python-aiplatform/issues/new/choose) before writing your code! That way we can discuss the change, evaluate designs, and agree on the general idea - [ ] Ensure the tests and linter pass - [x] Code coverage does not decrease (if any source code was changed) - [x] Appropriate docs were updated (if necessary) Fixes #192254729 🦕
diff --git a/samples/model-builder/create_training_pipeline_tabular_regression_sample.py b/samples/model-builder/create_training_pipeline_tabular_regression_sample.py
@@ -33,6 +33,7 @@ def create_training_pipeline_tabular_regression_sample(
 
     tabular_regression_job = aiplatform.AutoMLTabularTrainingJob(
         display_name=display_name,
+        optimization_prediction_type="regression"
     )
 
     my_tabular_dataset = aiplatform.TabularDataset(dataset_id)
diff --git a/samples/model-builder/create_training_pipeline_tabular_regression_sample_test.py b/samples/model-builder/create_training_pipeline_tabular_regression_sample_test.py
@@ -44,6 +44,7 @@ def test_create_training_pipeline_tabular_regression_sample(
     )
     mock_get_automl_tabular_training_job.assert_called_once_with(
         display_name=constants.DISPLAY_NAME,
+        optimization_prediction_type="regression"
     )
     mock_run_automl_tabular_training_job.assert_called_once_with(
         dataset=mock_tabular_dataset,

Original file line number	Diff line number	Diff line change
`@@ -33,6 +33,7 @@ def create_training_pipeline_tabular_regression_sample(`
`33`	`33`
`34`	`34`	`tabular_regression_job = aiplatform.AutoMLTabularTrainingJob(`
`35`	`35`	`display_name=display_name,`
	`36`	`+ optimization_prediction_type="regression"`
`36`	`37`	`)`
`37`	`38`
`38`	`39`	`my_tabular_dataset = aiplatform.TabularDataset(dataset_id)`
Original file line number	Diff line number	Diff line change
`@@ -44,6 +44,7 @@ def test_create_training_pipeline_tabular_regression_sample(`
`44`	`44`	`)`
`45`	`45`	`mock_get_automl_tabular_training_job.assert_called_once_with(`
`46`	`46`	`display_name=constants.DISPLAY_NAME,`
	`47`	`+ optimization_prediction_type="regression"`
`47`	`48`	`)`
`48`	`49`	`mock_run_automl_tabular_training_job.assert_called_once_with(`
`49`	`50`	`dataset=mock_tabular_dataset,`