Black formatter

Author: jprakash-db

src/databricks/sql/client.py

@@ -744,7 +744,7 @@ def execute(
         self,
         operation: str,
         parameters: Optional[TParameterCollection] = None,
-        enforce_embedded_schema_correctness=False
+        enforce_embedded_schema_correctness=False,
     ) -> "Cursor":
         """
         Execute a query and wait for execution to complete.
@@ -809,7 +809,7 @@ def execute(
             use_cloud_fetch=self.connection.use_cloud_fetch,
             parameters=prepared_params,
             async_op=False,
-            enforce_embedded_schema_correctness=enforce_embedded_schema_correctness
+            enforce_embedded_schema_correctness=enforce_embedded_schema_correctness,
         )
         self.active_result_set = ResultSet(
             self.connection,
@@ -831,7 +831,7 @@ def execute_async(
         self,
         operation: str,
         parameters: Optional[TParameterCollection] = None,
-        enforce_embedded_schema_correctness=False
+        enforce_embedded_schema_correctness=False,
     ) -> "Cursor":
         """
 
@@ -872,7 +872,7 @@ def execute_async(
             use_cloud_fetch=self.connection.use_cloud_fetch,
             parameters=prepared_params,
             async_op=True,
-            enforce_embedded_schema_correctness=enforce_embedded_schema_correctness
+            enforce_embedded_schema_correctness=enforce_embedded_schema_correctness,
         )
 
         return self

src/databricks/sql/thrift_backend.py

@@ -883,7 +883,7 @@ def execute_command(
         use_cloud_fetch=True,
         parameters=[],
         async_op=False,
-        enforce_embedded_schema_correctness=False
+        enforce_embedded_schema_correctness=False,
     ):
         assert session_handle is not None
 
@@ -915,7 +915,7 @@ def execute_command(
             },
             useArrowNativeTypes=spark_arrow_types,
             parameters=parameters,
-            enforceEmbeddedSchemaCorrectness=enforce_embedded_schema_correctness
+            enforceEmbeddedSchemaCorrectness=enforce_embedded_schema_correctness,
         )
         resp = self.make_request(self._client.ExecuteStatement, req)
 

tests/e2e/test_driver.py

@@ -880,4 +880,4 @@ def test_initial_namespace(self):
             cursor.execute("select current_catalog()")
             assert cursor.fetchone()[0] == self.arguments["catalog"]
             cursor.execute("select current_database()")
-            assert cursor.fetchone()[0] == table_name
+            assert cursor.fetchone()[0] == table_name