From 6d9b7d681ec8516488cfa4f73f7a44b3135eeebf Mon Sep 17 00:00:00 2001 From: Mingdao Yang Date: Sun, 13 Jul 2025 13:04:32 +0800 Subject: [PATCH] Cleanup type ignores --- .../sql/src/airflow/providers/common/sql/operators/sql.py | 2 +- .../common/sql/tests/unit/common/sql/sensors/test_sql.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/providers/common/sql/src/airflow/providers/common/sql/operators/sql.py b/providers/common/sql/src/airflow/providers/common/sql/operators/sql.py index fa6963c86e235..95d602bc9b46c 100644 --- a/providers/common/sql/src/airflow/providers/common/sql/operators/sql.py +++ b/providers/common/sql/src/airflow/providers/common/sql/operators/sql.py @@ -1247,7 +1247,7 @@ def execute(self, context: Context): ) # TODO(potiuk) remove the type ignore once we solve provider <-> Task SDK relationship - self.skip_all_except(context["ti"], follow_branch) # type: ignore[arg-type] + self.skip_all_except(context["ti"], follow_branch) def _initialize_partition_clause(clause: str | None) -> str | None: diff --git a/providers/common/sql/tests/unit/common/sql/sensors/test_sql.py b/providers/common/sql/tests/unit/common/sql/sensors/test_sql.py index d55fc22a27f79..b1d16ed1bef2c 100644 --- a/providers/common/sql/tests/unit/common/sql/sensors/test_sql.py +++ b/providers/common/sql/tests/unit/common/sql/sensors/test_sql.py @@ -226,7 +226,7 @@ def test_sql_sensor_postgres_poke_invalid_failure(self, mock_hook): task_id="sql_sensor_check", conn_id="postgres_default", sql="SELECT 1", - failure=[1], # type: ignore[arg-type] + failure=[1], ) mock_hook.get_connection.return_value.get_hook.return_value = mock.MagicMock(spec=DbApiHook) @@ -246,7 +246,7 @@ def test_sql_sensor_postgres_poke_invalid_success( task_id="sql_sensor_check", conn_id="postgres_default", sql="SELECT 1", - success=[1], # type: ignore[arg-type] + success=[1], ) mock_hook.get_connection.return_value.get_hook.return_value = mock.MagicMock(spec=DbApiHook)