From 0eeeb7281e6b5c1fc09c39a469f12f0d353424a5 Mon Sep 17 00:00:00 2001 From: Ryota Egashira Date: Mon, 3 Mar 2025 12:06:50 -0800 Subject: [PATCH] minor change --- .../src/datahub/ingestion/source/vertexai.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/metadata-ingestion/src/datahub/ingestion/source/vertexai.py b/metadata-ingestion/src/datahub/ingestion/source/vertexai.py index fca1248f075fbf..2e0255f1158a9f 100644 --- a/metadata-ingestion/src/datahub/ingestion/source/vertexai.py +++ b/metadata-ingestion/src/datahub/ingestion/source/vertexai.py @@ -371,8 +371,8 @@ def _get_job_output_workunits( model_version = self._search_model_version(model, model_version_str) if model and model_version: logger.info( - f" found a training job: {job.display_name} generated " - f"a model (name:{model.display_name} id:{model_version_str})" + f"Found output model (name:{model.display_name} id:{model_version_str}) " + f"for training job: {job.display_name}" ) yield from self._gen_ml_model_endpoint_workunits( model, model_version, job_urn @@ -486,9 +486,13 @@ def _gen_input_dataset_workunits( aspect=DataProcessInstanceInputClass(inputs=[dataset_urn]), ) logger.info( - f"Found input dataset ({dataset_name}) for training job ({job.display_name})" + f"Found the name of input dataset ({dataset_name}) with dataset id ({dataset_id})" ) yield from auto_workunit([mcp]) + else: + logger.error( + f"Unable to find the name of input dataset ({dataset_name}) with dataset id ({dataset_id})" + ) def _gen_endpoint_workunits( self, endpoint: Endpoint, model: Model, model_version: VersionInfo