docs: Add probabilistic inference to TiDE and L2L model code samples.

PiperOrigin-RevId: 570445202

Author: vertex-sdk-bot

samples/model-builder/create_training_pipeline_forecasting_sample.py

@@ -44,6 +44,7 @@ def create_training_pipeline_forecasting_sample(
     export_evaluated_data_items_bigquery_destination_uri: Optional[str] = None,
     export_evaluated_data_items_override_destination: bool = False,
     quantiles: Optional[List[float]] = None,
+    enable_probabilistic_inference: bool = False,
     validation_options: Optional[str] = None,
     predefined_split_column_name: Optional[str] = None,
     sync: bool = True,
@@ -81,6 +82,7 @@ def create_training_pipeline_forecasting_sample(
         export_evaluated_data_items_bigquery_destination_uri=export_evaluated_data_items_bigquery_destination_uri,
         export_evaluated_data_items_override_destination=export_evaluated_data_items_override_destination,
         quantiles=quantiles,
+        enable_probabilistic_inference=enable_probabilistic_inference,
         validation_options=validation_options,
         budget_milli_node_hours=budget_milli_node_hours,
         model_display_name=model_display_name,

samples/model-builder/create_training_pipeline_forecasting_sample_test.py

@@ -43,6 +43,7 @@ def test_create_training_pipeline_forecasting_sample(
         export_evaluated_data_items_bigquery_destination_uri=constants.EXPORT_EVALUATED_DATA_ITEMS_BIGQUERY_DESTINATION_URI,
         export_evaluated_data_items_override_destination=constants.EXPORT_EVALUATED_DATA_ITEMS_OVERRIDE_DESTINATION,
         quantiles=constants.QUANTILES,
+        enable_probabilistic_inference=constants.ENABLE_PROBABILISTIC_INFERENCE,
         validation_options=constants.VALIDATION_OPTIONS,
         predefined_split_column_name=constants.PREDEFINED_SPLIT_COLUMN_NAME,
     )
@@ -79,6 +80,7 @@ def test_create_training_pipeline_forecasting_sample(
         export_evaluated_data_items_bigquery_destination_uri=constants.EXPORT_EVALUATED_DATA_ITEMS_BIGQUERY_DESTINATION_URI,
         export_evaluated_data_items_override_destination=constants.EXPORT_EVALUATED_DATA_ITEMS_OVERRIDE_DESTINATION,
         quantiles=constants.QUANTILES,
+        enable_probabilistic_inference=constants.ENABLE_PROBABILISTIC_INFERENCE,
         validation_options=constants.VALIDATION_OPTIONS,
         predefined_split_column_name=constants.PREDEFINED_SPLIT_COLUMN_NAME,
         sync=True,

samples/model-builder/create_training_pipeline_forecasting_tide_sample.py

@@ -44,6 +44,7 @@ def create_training_pipeline_forecasting_time_series_dense_encoder_sample(
     export_evaluated_data_items_bigquery_destination_uri: Optional[str] = None,
     export_evaluated_data_items_override_destination: bool = False,
     quantiles: Optional[List[float]] = None,
+    enable_probabilistic_inference: bool = False,
     validation_options: Optional[str] = None,
     predefined_split_column_name: Optional[str] = None,
     sync: bool = True,
@@ -82,6 +83,7 @@ def create_training_pipeline_forecasting_time_series_dense_encoder_sample(
         export_evaluated_data_items_bigquery_destination_uri=export_evaluated_data_items_bigquery_destination_uri,
         export_evaluated_data_items_override_destination=export_evaluated_data_items_override_destination,
         quantiles=quantiles,
+        enable_probabilistic_inference=enable_probabilistic_inference,
         validation_options=validation_options,
         budget_milli_node_hours=budget_milli_node_hours,
         model_display_name=model_display_name,

samples/model-builder/create_training_pipeline_forecasting_tide_sample_test.py

@@ -43,6 +43,7 @@ def test_create_training_pipeline_forecasting_tide_sample(
         export_evaluated_data_items_bigquery_destination_uri=constants.EXPORT_EVALUATED_DATA_ITEMS_BIGQUERY_DESTINATION_URI,
         export_evaluated_data_items_override_destination=constants.EXPORT_EVALUATED_DATA_ITEMS_OVERRIDE_DESTINATION,
         quantiles=constants.QUANTILES,
+        enable_probabilistic_inference=constants.ENABLE_PROBABILISTIC_INFERENCE,
         validation_options=constants.VALIDATION_OPTIONS,
         predefined_split_column_name=constants.PREDEFINED_SPLIT_COLUMN_NAME,
     )
@@ -79,6 +80,7 @@ def test_create_training_pipeline_forecasting_tide_sample(
         export_evaluated_data_items_bigquery_destination_uri=constants.EXPORT_EVALUATED_DATA_ITEMS_BIGQUERY_DESTINATION_URI,
         export_evaluated_data_items_override_destination=constants.EXPORT_EVALUATED_DATA_ITEMS_OVERRIDE_DESTINATION,
         quantiles=constants.QUANTILES,
+        enable_probabilistic_inference=constants.ENABLE_PROBABILISTIC_INFERENCE,
         validation_options=constants.VALIDATION_OPTIONS,
         predefined_split_column_name=constants.PREDEFINED_SPLIT_COLUMN_NAME,
         sync=True,

samples/model-builder/test_constants.py

@@ -269,6 +269,7 @@
 EXPORT_EVALUATED_DATA_ITEMS_BIGQUERY_DESTINATION_URI = "bq://test:test:test"
 EXPORT_EVALUATED_DATA_ITEMS_OVERRIDE_DESTINATION = True
 QUANTILES = [0, 0.5, 1]
+ENABLE_PROBABILISTIC_INFERENCE = True
 VALIDATION_OPTIONS = "fail-pipeline"
 PREDEFINED_SPLIT_COLUMN_NAME = "predefined"