[airflow] branch main updated: Fix formatting and missing import (#16455)

[as...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

ash pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/main by this push:
     new 608dd0d  Fix formatting and missing import (#16455)
608dd0d is described below

commit 608dd0ddf65dac7f7eee2cb54628a93805b7ad66
Author: Ash Berlin-Taylor <as...@firemirror.com>
AuthorDate: Tue Jun 15 12:11:59 2021 +0100

    Fix formatting and missing import (#16455)
---
 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 6febfcc..0bb9bc5 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 @@ class SnowflakeHook(DbApiHook):
             "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