feat: enabling AutoML Forecasting training response to include BigQuery location of exported evaluated examples

google/cloud/aiplatform/training_jobs.py

@@ -4119,7 +4119,7 @@ def _run(
             labels=model_labels or self._labels,
         )
 
-        return self._run_job(
+        new_model = self._run_job(
             training_task_definition=training_task_definition,
             training_task_inputs=training_task_inputs_dict,
             dataset=dataset,
@@ -4130,6 +4130,14 @@ def _run(
             model=model,
         )
 
+        if export_evaluated_data_items:
+            _LOGGER.info(
+                "Exported examples available at:\n%s"
+                % self.evaluated_data_items_bigquery_uri
+            )
+
+        return new_model
+
     @property
     def _model_upload_fail_string(self) -> str:
         """Helper property for model upload failure."""
@@ -4138,6 +4146,20 @@ def _model_upload_fail_string(self) -> str:
             "Model."
         )
 
+    @property
+    def evaluated_data_items_bigquery_uri(self) -> Optional[str]:
+        """BigQuery location of exported evaluated examples from the Training Job"""
+
+        self._assert_gca_resource_is_available()
+
+        try:
+            metadata = self._gca_resource.training_task_metadata
+            return metadata["evaluatedDataItemsBigqueryUri"]
+        except (AttributeError, KeyError):
+            raise ValueError(
+                "BigQuery URI for evaluated data items does not exist. Must export evaluated data items during training."
+            )
+
     def _add_additional_experiments(self, additional_experiments: List[str]):
         """Add experiment flags to the training job.
         Args: