From 4906d9138376965bf87efc7875d366a75142ac29 Mon Sep 17 00:00:00 2001 From: anand Date: Thu, 29 May 2025 10:12:12 -0500 Subject: [PATCH 1/2] change hasattr to get --- .../src/airflow/providers/openlineage/plugins/macros.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/providers/openlineage/src/airflow/providers/openlineage/plugins/macros.py b/providers/openlineage/src/airflow/providers/openlineage/plugins/macros.py index 5fd47b418c9b7..df0375d3cf772 100644 --- a/providers/openlineage/src/airflow/providers/openlineage/plugins/macros.py +++ b/providers/openlineage/src/airflow/providers/openlineage/plugins/macros.py @@ -127,7 +127,7 @@ def _get_logical_date(task_instance): context = task_instance.get_template_context() if hasattr(task_instance, "dag_run"): dag_run = task_instance.dag_run - elif hasattr(context, "dag_run"): + elif context.get("dag_run"): dag_run = context["dag_run"] if hasattr(dag_run, "logical_date") and dag_run.logical_date: date = dag_run.logical_date From 1d13c4a986899cf4c7ec1873c865910e657865e2 Mon Sep 17 00:00:00 2001 From: anand Date: Thu, 29 May 2025 10:52:25 -0500 Subject: [PATCH 2/2] update to else --- .../src/airflow/providers/openlineage/plugins/macros.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/providers/openlineage/src/airflow/providers/openlineage/plugins/macros.py b/providers/openlineage/src/airflow/providers/openlineage/plugins/macros.py index df0375d3cf772..8cae51cb744e5 100644 --- a/providers/openlineage/src/airflow/providers/openlineage/plugins/macros.py +++ b/providers/openlineage/src/airflow/providers/openlineage/plugins/macros.py @@ -127,7 +127,7 @@ def _get_logical_date(task_instance): context = task_instance.get_template_context() if hasattr(task_instance, "dag_run"): dag_run = task_instance.dag_run - elif context.get("dag_run"): + else: dag_run = context["dag_run"] if hasattr(dag_run, "logical_date") and dag_run.logical_date: date = dag_run.logical_date