Skip to content

Commit 3ec620c

Browse files
fix: issues/192254729 (#914)
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 🦕
1 parent 0f22ff6 commit 3ec620c

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

samples/model-builder/create_training_pipeline_tabular_regression_sample.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ def create_training_pipeline_tabular_regression_sample(
3333

3434
tabular_regression_job = aiplatform.AutoMLTabularTrainingJob(
3535
display_name=display_name,
36+
optimization_prediction_type="regression"
3637
)
3738

3839
my_tabular_dataset = aiplatform.TabularDataset(dataset_id)

samples/model-builder/create_training_pipeline_tabular_regression_sample_test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ def test_create_training_pipeline_tabular_regression_sample(
4444
)
4545
mock_get_automl_tabular_training_job.assert_called_once_with(
4646
display_name=constants.DISPLAY_NAME,
47+
optimization_prediction_type="regression"
4748
)
4849
mock_run_automl_tabular_training_job.assert_called_once_with(
4950
dataset=mock_tabular_dataset,

0 commit comments

Comments
 (0)