diff --git a/astronomer/providers/core/sensors/astro.py b/astronomer/providers/core/sensors/astro.py index a9b90494c..bff4f68ed 100644 --- a/astronomer/providers/core/sensors/astro.py +++ b/astronomer/providers/core/sensors/astro.py @@ -62,7 +62,7 @@ def poke(self, context: Context) -> bool | PokeReturnValue: if dag_runs is None or len(dag_runs) == 0: self.log.info(f"No dag-runs found for dag-id {self.external_dag_id}") return True - self._dag_run_id = dag_runs[0].get("dag_run_id") + self._dag_run_id: str = dag_runs[0]["dag_run_id"] if self.external_task_id is not None: task_instance = hook.get_task_instance( self.external_dag_id, self._dag_run_id, self.external_task_id