Fix linting in src for black

Author: samikshya-db

src/databricks/sql/auth/auth.py

@@ -120,9 +120,11 @@ def get_python_sql_connector_auth_provider(hostname: str, **kwargs):
         tls_client_cert_file=kwargs.get("_tls_client_cert_file"),
         oauth_scopes=PYSQL_OAUTH_SCOPES,
         oauth_client_id=kwargs.get("oauth_client_id") or client_id,
-        oauth_redirect_port_range=[kwargs["oauth_redirect_port"]]
-        if kwargs.get("oauth_client_id") and kwargs.get("oauth_redirect_port")
-        else redirect_port_range,
+        oauth_redirect_port_range=(
+            [kwargs["oauth_redirect_port"]]
+            if kwargs.get("oauth_client_id") and kwargs.get("oauth_redirect_port")
+            else redirect_port_range
+        ),
         oauth_persistence=kwargs.get("experimental_oauth_persistence"),
         credentials_provider=kwargs.get("credentials_provider"),
     )

src/databricks/sql/auth/authenticators.py

@@ -25,12 +25,10 @@ class CredentialsProvider(abc.ABC):
     for authenticating requests to Databricks REST APIs"""
 
     @abc.abstractmethod
-    def auth_type(self) -> str:
-        ...
+    def auth_type(self) -> str: ...
 
     @abc.abstractmethod
-    def __call__(self, *args, **kwargs) -> HeaderFactory:
-        ...
+    def __call__(self, *args, **kwargs) -> HeaderFactory: ...
 
 
 # Private API: this is an evolving interface and it will change in the future.

src/databricks/sql/auth/thrift_http_client.py

@@ -115,9 +115,11 @@ def open(self):
             pool_class = HTTPSConnectionPool
             _pool_kwargs.update(
                 {
-                    "cert_reqs": ssl.CERT_REQUIRED
-                    if self._ssl_options.tls_verify
-                    else ssl.CERT_NONE,
+                    "cert_reqs": (
+                        ssl.CERT_REQUIRED
+                        if self._ssl_options.tls_verify
+                        else ssl.CERT_NONE
+                    ),
                     "ca_certs": self._ssl_options.tls_trusted_ca_file,
                     "cert_file": self._ssl_options.tls_client_cert_file,
                     "key_file": self._ssl_options.tls_client_cert_key_file,

src/databricks/sql/cloudfetch/downloader.py

@@ -100,7 +100,7 @@ def run(self) -> DownloadedFile:
                 self.link.fileLink,
                 timeout=self.settings.download_timeout,
                 verify=self._ssl_options.tls_verify,
-                headers=self.link.httpHeaders
+                headers=self.link.httpHeaders,
                 # TODO: Pass cert from `self._ssl_options`
             )
             response.raise_for_status()

src/databricks/sql/thrift_backend.py

@@ -613,7 +613,10 @@ def _create_arrow_table(self, t_row_set, lz4_compressed, schema_bytes, descripti
                 num_rows,
             ) = convert_column_based_set_to_arrow_table(t_row_set.columns, description)
         elif t_row_set.arrowBatches is not None:
-            (arrow_table, num_rows,) = convert_arrow_based_set_to_arrow_table(
+            (
+                arrow_table,
+                num_rows,
+            ) = convert_arrow_based_set_to_arrow_table(
                 t_row_set.arrowBatches, lz4_compressed, schema_bytes
             )
         else: