From b460f7769fcff6111b771d7251fda8cec001e8ff Mon Sep 17 00:00:00 2001 From: Pankaj Date: Wed, 7 Feb 2024 11:21:39 +0530 Subject: [PATCH] Fix mypy --- astronomer/providers/core/triggers/astro.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/astronomer/providers/core/triggers/astro.py b/astronomer/providers/core/triggers/astro.py index 20bc75254..11a8693bd 100644 --- 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)