From faece5be98dd5f5b0b2406aa5bd58543df981068 Mon Sep 17 00:00:00 2001 From: Wei Lee Date: Mon, 27 Nov 2023 12:09:43 +0800 Subject: [PATCH] style(snowflake): fix mypy error --- astronomer/providers/snowflake/operators/snowflake.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/astronomer/providers/snowflake/operators/snowflake.py b/astronomer/providers/snowflake/operators/snowflake.py index 5cc6e9d24..dfab58de2 100644 --- a/astronomer/providers/snowflake/operators/snowflake.py +++ b/astronomer/providers/snowflake/operators/snowflake.py @@ -181,7 +181,7 @@ def execute(self, context: Context) -> None: self.log.info("SQL after adding query tag: %s", self.sql) hook = self.get_db_hook() - hook.run(self.sql, parameters=self.parameters) # type: ignore[arg-type] + hook.run(self.sql, parameters=self.parameters) self.query_ids = hook.query_ids if self.do_xcom_push: @@ -350,9 +350,7 @@ def execute(self, context: Context) -> None: token_life_time=self.token_life_time, token_renewal_delta=self.token_renewal_delta, ) - hook.execute_query( - self.sql, statement_count=self.statement_count, bindings=self.bindings # type: ignore[arg-type] - ) + hook.execute_query(self.sql, statement_count=self.statement_count, bindings=self.bindings) self.query_ids = hook.query_ids self.log.info("List of query ids %s", self.query_ids)