Fixes for databricks

Author: erezsh

data_diff/databases/database_types.py

@@ -170,11 +170,10 @@ def select_table_schema(self, path: DbPath) -> str:
         "Provide SQL for selecting the table schema as (name, type, date_prec, num_prec)"
         ...
 
-
     @abstractmethod
     def query_table_schema(self, path: DbPath) -> Dict[str, tuple]:
         """Query the table for its schema for table in 'path', and return {column: tuple}
-          where the tuple is (table_name, col_name, type_repr, datetime_precision?, numeric_precision?, numeric_scale?)
+        where the tuple is (table_name, col_name, type_repr, datetime_precision?, numeric_precision?, numeric_scale?)
         """
         ...
 

data_diff/databases/databricks.py

@@ -80,11 +80,10 @@ def query_table_schema(self, path: DbPath) -> Dict[str, tuple]:
             if not rows:
                 raise RuntimeError(f"{self.name}: Table '{'.'.join(path)}' does not exist, or has no columns")
 
-            d = {r[0]: r for r in rows}
+            d = {r.COLUMN_NAME: r for r in rows}
             assert len(d) == len(rows)
             return d
 
-
     def _process_table_schema(self, path: DbPath, raw_schema: Dict[str, tuple], filter_columns: Sequence[str]):
         accept = {i.lower() for i in filter_columns}
         rows = [row for name, row in raw_schema.items() if name.lower() in accept]