Add new type of exception to catch timeout (apache#42064)

potiuk · Sep 6, 2024 · abe5f7c · abe5f7c
1 parent c7882a3
commit abe5f7c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/airflow/models/dagbag.py b/airflow/models/dagbag.py
@@ -46,6 +46,7 @@
     AirflowDagCycleException,
     AirflowDagDuplicatedIdException,
     AirflowException,
+    AirflowTaskTimeout,
 )
 from airflow.listeners.listener import get_listener_manager
 from airflow.models.base import Base
@@ -350,7 +351,7 @@ def parse(mod_name, filepath):
                 sys.modules[spec.name] = new_module
                 loader.exec_module(new_module)
                 return [new_module]
-            except Exception as e:
+            except (Exception, AirflowTaskTimeout) as e:
                 DagContext.autoregistered_dags.clear()
                 self.log.exception("Failed to import: %s", filepath)
                 if self.dagbag_import_error_tracebacks: