style: fix mypy warning (#1541)

astronomer · Jun 3, 2024 · 45983b6 · 45983b6
1 parent 9714727
commit 45983b6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/astronomer/providers/snowflake/example_dags/example_snowflake_sensor.py b/astronomer/providers/snowflake/example_dags/example_snowflake_sensor.py
@@ -67,7 +67,7 @@
         sql=SNOWFLAKE_SLACK_SQL,
         poke_interval=POKE_INTERVAL,
         timeout=TASK_TIMEOUT * 60,
-        success=lambda result: True,  # type: ignore[arg-type]
+        success=lambda result: True,
     )
 
     snowflake_op_sql_str >> snowflake_op_with_params >> snowflake_op_sql_sensor >> snowflake_with_callable
diff --git a/astronomer/providers/snowflake/sensors/snowflake.py b/astronomer/providers/snowflake/sensors/snowflake.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from datetime import timedelta
-from typing import Any, Sequence
+from typing import Any, Callable, Sequence
 
 from airflow.exceptions import AirflowException
 from airflow.providers.common.sql.sensors.sql import SqlSensor
@@ -48,8 +48,8 @@ def __init__(
         snowflake_conn_id: str,
         sql: str,
         parameters: dict[str, Any] | None = None,
-        success: str | None = None,
-        failure: str | None = None,
+        success: Callable[[Any], bool] | None = None,
+        failure: Callable[[Any], bool] | None = None,
         fail_on_empty: bool = False,
         hook_params: dict[str, Any] | None = None,
         **kwargs: Any,