fix: pass key columns

Apologies, I only tested the case where --stats is passed.

Author: stefankeidel

data_diff/format.py

@@ -60,7 +60,7 @@ def jsonify(
     t1_exclusive_rows, t2_exclusive_rows, diff_rows = _group_rows(diff_info, schema)
 
     if not stats_only:
-        rows = _make_rows_diff(t1_exclusive_rows, t2_exclusive_rows, diff_rows)
+        rows = _make_rows_diff(t1_exclusive_rows, t2_exclusive_rows, diff_rows, key_columns)
 
     summary = None
     if with_summary:
@@ -252,7 +252,8 @@ def _group_rows(
 def _make_rows_diff(
     t1_exclusive_rows: List[Dict[str, Any]],
     t2_exclusive_rows: List[Dict[str, Any]],
-    diff_rows: List[Dict[str, Any]]
+    diff_rows: List[Dict[str, Any]],
+    key_columns: List[str],
 ) -> RowsDiff:
     diff_rows_jsonified = []
     for row in diff_rows: