From 82fcbb79ac7098c98321ccb6f6f6341ecb634258 Mon Sep 17 00:00:00 2001 From: Harry Plumer Date: Tue, 9 Nov 2021 22:43:52 -0700 Subject: [PATCH] Remove duplicate get_connection in SnowflakeHook --- airflow/providers/snowflake/hooks/snowflake.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/airflow/providers/snowflake/hooks/snowflake.py b/airflow/providers/snowflake/hooks/snowflake.py index 34596e1617674..8b026b5b5ee8c 100644 --- a/airflow/providers/snowflake/hooks/snowflake.py +++ b/airflow/providers/snowflake/hooks/snowflake.py @@ -268,8 +268,7 @@ def run(self, sql: Union[str, list], autocommit: bool = False, parameters: Optio """ self.query_ids = [] - with self.get_conn() as conn: - conn = self.get_conn() + with closing(self.get_conn()) as conn: self.set_autocommit(conn, autocommit) if isinstance(sql, str):