Merge pull request #222 from DataRecce/feature/drc-285-table-schema-a…

…re-not-parsed-properly [Bug] Fix the schema diff column name is wrong
DataRecce · Mar 11, 2024 · 40a8275 · 40a8275
2 parents 2226c10 + 5a21906
commit 40a8275
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/recce/dbt.py b/recce/dbt.py
@@ -313,7 +313,7 @@ def get_lineage(self, base: Optional[bool] = False):
                     cols_unique.append(child_name[len(unique_prefix):])
 
             if catalog is not None and unique_id in catalog.nodes:
-                columns = []
+                columns = {}
                 primary_key = None
                 for col_name, col in catalog.nodes[unique_id].columns.items():
                     col = dict(name=col_name, type=col.type)
@@ -323,7 +323,7 @@ def get_lineage(self, base: Optional[bool] = False):
                         col['unique'] = True
                         if not primary_key:
                             primary_key = col_name
-                    columns.append(col)
+                    columns[col_name] = col
                 nodes[unique_id]['columns'] = columns
                 if primary_key:
                     nodes[unique_id]['primary_key'] = primary_key