diff --git a/daft/dataframe/dataframe.py b/daft/dataframe/dataframe.py index 3c5ac1b066..0e841a5196 100644 --- a/daft/dataframe/dataframe.py +++ b/daft/dataframe/dataframe.py @@ -2515,7 +2515,7 @@ def to_arrow(self, cast_tensors_to_ray_tensor_dtype: bool = False) -> "pyarrow.T results_buffer_size=None, # Hardcoded for now, we can pass it through cast_tensors_to_ray_tensor_dtype=cast_tensors_to_ray_tensor_dtype, ) - return pa.Table.from_batches(arrow_rb_iter) + return pa.Table.from_batches(arrow_rb_iter, schema=self.schema().to_pyarrow_schema()) @DataframePublicAPI def to_pydict(self) -> Dict[str, List[Any]]: