fix: Missing keys in list_to_arrow_table (#2778)

aws · Apr 22, 2024 · 60beb95 · 60beb95
1 parent c8b7b03
commit 60beb95
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/awswrangler/_utils.py b/awswrangler/_utils.py
@@ -899,6 +899,10 @@ def list_to_arrow_table(
         names = []
         if mapping:
             names = list(mapping[0].keys())
+            for row in mapping:
+                for k in row.keys():
+                    if k not in names:
+                        names.append(k)
         for n in names:
             v = [row[n] if n in row else None for row in mapping]
             arrays.append(v)