From e3a99e2f840269ad5b9a11a6608fae48b5d7a92c Mon Sep 17 00:00:00 2001 From: Sriharsha Chintalapani Date: Wed, 10 May 2023 21:43:57 -0700 Subject: [PATCH] Fix Airflow ingestion: check if downstream task are null (#11539) * Fix Airflow ingestion: check if downstream task are null * Fix Airflow ingestion: check if downstream task are null --- .../metadata/ingestion/source/pipeline/airflow/metadata.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ingestion/src/metadata/ingestion/source/pipeline/airflow/metadata.py b/ingestion/src/metadata/ingestion/source/pipeline/airflow/metadata.py index 6d66c414ad3..fdfd7dff804 100644 --- a/ingestion/src/metadata/ingestion/source/pipeline/airflow/metadata.py +++ b/ingestion/src/metadata/ingestion/source/pipeline/airflow/metadata.py @@ -313,7 +313,9 @@ class AirflowSource(PipelineServiceSource): f"{clean_uri(host_port)}/taskinstance/list/" f"?flt1_dag_id_equals={dag.dag_id}&_flt_3_task_id={task.task_id}" ), - downstreamTasks=list(task.downstream_task_ids), + downstreamTasks=list(task.downstream_task_ids) + if task.downstream_task_ids + else [], startDate=task.start_date.isoformat() if task.start_date else None, endDate=task.end_date.isoformat() if task.end_date else None, )