From d06c14f52757321f2049bb54212421f68bf3ed06 Mon Sep 17 00:00:00 2001 From: Sam Wheating Date: Thu, 24 Aug 2023 00:22:04 -0700 Subject: [PATCH] Set snowflake_conn_id on Snowflake Operators to avoid error (#33681) --- airflow/providers/snowflake/operators/snowflake.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/airflow/providers/snowflake/operators/snowflake.py b/airflow/providers/snowflake/operators/snowflake.py index 090f9cf384293..864247dda1097 100644 --- a/airflow/providers/snowflake/operators/snowflake.py +++ b/airflow/providers/snowflake/operators/snowflake.py @@ -213,6 +213,7 @@ def __init__( session_parameters: dict | None = None, **kwargs, ) -> None: + self.snowflake_conn_id = snowflake_conn_id if any([warehouse, database, role, schema, authenticator, session_parameters]): hook_params = kwargs.pop("hook_params", {}) kwargs["hook_params"] = { @@ -282,6 +283,7 @@ def __init__( session_parameters: dict | None = None, **kwargs, ) -> None: + self.snowflake_conn_id = snowflake_conn_id if any([warehouse, database, role, schema, authenticator, session_parameters]): hook_params = kwargs.pop("hook_params", {}) kwargs["hook_params"] = { @@ -362,6 +364,7 @@ def __init__( session_parameters: dict | None = None, **kwargs, ) -> None: + self.snowflake_conn_id = snowflake_conn_id if any([warehouse, database, role, schema, authenticator, session_parameters]): hook_params = kwargs.pop("hook_params", {}) kwargs["hook_params"] = {