fix(airflow): fix lint related to dag_run field (datahub-project#11616)

skrydal · Oct 15, 2024 · be1b880 · be1b880
1 parent 0d06a61
commit be1b880
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/metadata-ingestion-modules/airflow-plugin/src/datahub_airflow_plugin/datahub_listener.py b/metadata-ingestion-modules/airflow-plugin/src/datahub_airflow_plugin/datahub_listener.py
@@ -383,13 +383,11 @@ def on_task_instance_running(
             return
 
         logger.debug(
-            f"DataHub listener got notification about task instance start for {task_instance.task_id} of dag {task_instance.dag_run.dag_id}"
+            f"DataHub listener got notification about task instance start for {task_instance.task_id} of dag {task_instance.dag_id}"
         )
 
-        if not self.config.dag_filter_pattern.allowed(task_instance.dag_run.dag_id):
-            logger.debug(
-                f"DAG {task_instance.dag_run.dag_id} is not allowed by the pattern"
-            )
+        if not self.config.dag_filter_pattern.allowed(task_instance.dag_id):
+            logger.debug(f"DAG {task_instance.dag_id} is not allowed by the pattern")
             return
 
         if self.config.render_templates: