Fixing final failures in linting reported on 'pre-commit' checks on l…

…ocal repo
Eventual-Inc · Oct 10, 2024 · 27bba26 · 27bba26
1 parent ffc9776
commit 27bba26
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/daft/unity_catalog/unity_catalog.py b/daft/unity_catalog/unity_catalog.py
@@ -97,7 +97,7 @@ def load_table(self, table_name: str) -> UnityCatalogTable:
 
         # Grab credentials from Unity catalog and place it into the Table
         temp_table_credentials = self._client.temporary_table_credentials.create(operation="READ", table_id=table_id)
-        
+
         scheme = urlparse(storage_location).scheme
         if scheme == "s3" or scheme == "s3a":
             aws_temp_credentials = temp_table_credentials.aws_temp_credentials
@@ -117,10 +117,8 @@ def load_table(self, table_name: str) -> UnityCatalogTable:
             pass
         elif scheme == "az" or scheme == "abfs" or scheme == "abfss":
             io_config = IOConfig(
-                    azure=AzureConfig(
-                        sas_token = temp_table_credentials.azure_user_delegation_sas.get('sas_token')
-                    )
-                )
+                azure=AzureConfig(sas_token=temp_table_credentials.azure_user_delegation_sas.get("sas_token"))
+            )
 
         return UnityCatalogTable(
             table_uri=storage_location,