fix tests

apache · Apr 27, 2024 · 9cfe9a5 · 9cfe9a5
1 parent a166e8d
commit 9cfe9a5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/airflow/models/taskinstance.py b/airflow/models/taskinstance.py
@@ -3087,14 +3087,18 @@ def handle_failure(
         if TYPE_CHECKING:
             assert self.task
             assert self.task.dag
+        try:
+            fail_stop = self.task.dag.fail_stop
+        except Exception:
+            fail_stop = False
         _handle_failure(
             task_instance=self,
             error=error,
             session=session,
             test_mode=test_mode,
             context=context,
             force_fail=force_fail,
-            fail_stop=self.task.dag.fail_stop,
+            fail_stop=fail_stop,
         )
 
     def is_eligible_to_retry(self):

diff --git a/airflow/serialization/pydantic/taskinstance.py b/airflow/serialization/pydantic/taskinstance.py
@@ -279,14 +279,18 @@ def handle_failure(
         if TYPE_CHECKING:
             assert self.task
             assert self.task.dag
+        try:
+            fail_stop = self.task.dag.fail_stop
+        except Exception:
+            fail_stop = False
         _handle_failure(
             task_instance=self,
             error=error,
             session=session,
             test_mode=test_mode,
             context=context,
             force_fail=force_fail,
-            fail_stop=self.task.dag.fail_stop,
+            fail_stop=fail_stop,
         )
 
     def refresh_from_task(self, task: Operator, pool_override: str | None = None) -> None: