Set snowflake_conn_id on Snowflake Operators to avoid error (#33681)

apache · Aug 24, 2023 · d06c14f · d06c14f
1 parent 2dbb963
commit d06c14f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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"] = {