Update table.py

delta-io · Nov 24, 2023 · fc596b6 · fc596b6
1 parent d90adbb
commit fc596b6
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/python/deltalake/table.py b/python/deltalake/table.py
@@ -632,18 +632,18 @@ def merge(
         )
 
         if isinstance(source, pyarrow.RecordBatchReader):
-            source = convert_pyarrow_recordbatchreader(source, large_dtypes=False)
+            source = convert_pyarrow_recordbatchreader(source, large_dtypes=True)
         elif isinstance(source, pyarrow.RecordBatch):
             source = convert_pyarrow_recordbatch(
-                source, large_dtypes=False
-            )  # TODO(ion): set to True once MERGE uses logical plan
+                source, large_dtypes=True
+            )
         elif isinstance(source, pyarrow.Table):
-            source = convert_pyarrow_table(source, large_dtypes=False)
+            source = convert_pyarrow_table(source, large_dtypes=True)
         elif isinstance(source, ds.Dataset):
-            source = convert_pyarrow_dataset(source, large_dtypes=False)
+            source = convert_pyarrow_dataset(source, large_dtypes=True)
         elif isinstance(source, pandas.DataFrame):
             source = convert_pyarrow_table(
-                pyarrow.Table.from_pandas(source), large_dtypes=False
+                pyarrow.Table.from_pandas(source), large_dtypes=True
             )
         else:
             raise TypeError(