From 30cc45ddeb873749079ba7a0c767845f8b1fa930 Mon Sep 17 00:00:00 2001 From: Ash Berlin-Taylor Date: Tue, 15 Jun 2021 11:59:34 +0100 Subject: [PATCH] Fix formatting and missing import --- airflow/providers/snowflake/hooks/snowflake.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/airflow/providers/snowflake/hooks/snowflake.py b/airflow/providers/snowflake/hooks/snowflake.py index 6febfcc2fff81..0bb9bc57ed94e 100644 --- a/airflow/providers/snowflake/hooks/snowflake.py +++ b/airflow/providers/snowflake/hooks/snowflake.py @@ -15,6 +15,7 @@ # KIND, either express or implied. See the License for the # specific language governing permissions and limitations # under the License. +import os from contextlib import closing from io import StringIO from typing import Any, Dict, Optional, Tuple, Union @@ -180,7 +181,7 @@ def _get_conn_params(self) -> Dict[str, Optional[str]]: "authenticator": self.authenticator or authenticator, "session_parameters": self.session_parameters or session_parameters, # application is used to track origin of the requests - "application": os.environ.get("AIRFLOW_SNOWFLAKE_PARTNER", "AIRFLOW") + "application": os.environ.get("AIRFLOW_SNOWFLAKE_PARTNER", "AIRFLOW"), } # If private_key_file is specified in the extra json, load the contents of the file as a private