Fix mypy

astronomer · Feb 7, 2024 · cad8ba6 · cad8ba6
1 parent 4a00728
commit cad8ba6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/astronomer/providers/core/sensors/astro.py b/astronomer/providers/core/sensors/astro.py
@@ -2,7 +2,7 @@
 
 import datetime
 import time
-from typing import Any
+from typing import Any, cast
 
 from airflow.exceptions import AirflowException, AirflowSkipException
 from airflow.sensors.base import BaseSensorOperator, PokeReturnValue
@@ -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: str = dag_runs[0]["dag_run_id"]
+        self._dag_run_id = cast(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