Fix mypy

astronomer · Feb 7, 2024 · b460f77 · b460f77
1 parent cad8ba6
commit b460f77
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/astronomer/providers/core/triggers/astro.py b/astronomer/providers/core/triggers/astro.py
@@ -63,9 +63,9 @@ async def run(self) -> AsyncIterator[TriggerEvent]:
  self.external_dag_id, self.dag_run_id, self.external_task_id
  )
  state = task_instance.get("state") if task_instance else None
- if state == "success":
+ if state in ("success", "skipped"):
  yield TriggerEvent({"status": "done"})
- elif state == "failed":
+ elif state in ("failed", "upstream_failed"):
  yield TriggerEvent({"status": "failed"})
  else:
  dag_run = hook.get_dag_run(self.external_dag_id, self.dag_run_id)