[BUG] bugfix for empty partitions when writing out empty partitions (#…

…1814)
Eventual-Inc · Jan 23, 2024 · 854037e · 854037e
1 parent 966145b
commit 854037e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/daft/table/table_io.py b/daft/table/table_io.py
@@ -453,7 +453,11 @@ def file_visitor(written_file, i=i):
             **kwargs,
         )
 
-    data_dict: dict[str, Any] = {schema.column_names()[0]: visited_paths}
+    data_dict: dict[str, Any] = {
+        schema.column_names()[0]: Series.from_pylist(visited_paths, name=schema.column_names()[0]).cast(
+            DataType.string()
+        )
+    }
 
     if partition_values is not None:
         partition_idx_series = Series.from_pylist(partition_idx).cast(DataType.int64())