Skip to content

Commit 3389c32

Browse files
committed
adding run status logic
1 parent a6cf015 commit 3389c32

File tree

1 file changed

+2
-4
lines changed
  • metadata-ingestion/src/datahub/ingestion/source/vertexai

1 file changed

+2
-4
lines changed

metadata-ingestion/src/datahub/ingestion/source/vertexai/vertexai.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -342,16 +342,14 @@ def _gen_experiment_run_mcps(
342342
(
343343
DataProcessInstanceRunEventClass(
344344
status=DataProcessRunStatusClass.COMPLETE,
345-
timestampMillis=created_time,
345+
timestampMillis=created_time if created_time else 0, # None is not allowed, 0 as default value
346346
result=DataProcessInstanceRunResultClass(
347347
type=run_result_type,
348348
nativeResultType=self.platform,
349349
),
350-
durationMillis=duration,
350+
durationMillis=duration if duration else None,
351351
)
352352
if is_status_for_run_event_class(run_result_type)
353-
and duration
354-
and created_time
355353
else None
356354
),
357355
],

0 commit comments

Comments
 (0)