Fix TimeUnit.

Eventual-Inc · Feb 21, 2024 · 87221fc · 87221fc
1 parent 5ead3fc
commit 87221fc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/daft/daft.pyi b/daft/daft.pyi
@@ -224,6 +224,7 @@ class ParquetSourceConfig:
     """
     Configuration of a Parquet data source.
     """
+
     coerce_int96_timestamp_unit: PyTimeUnit | None
 
     def __init__(self, coerce_int96_timestamp_unit: PyTimeUnit | None = None): ...

diff --git a/daft/execution/execution_step.py b/daft/execution/execution_step.py
@@ -23,6 +23,7 @@
     ResourceRequest,
     StorageConfig,
 )
+from daft.datatype import TimeUnit
 from daft.expressions import Expression, ExpressionsProjection, col
 from daft.logical.map_partition_ops import MapPartitionOp
 from daft.logical.schema import Schema
@@ -409,7 +410,7 @@ def _handle_tabular_files_scan(
                         storage_config=self.storage_config,
                         read_options=read_options,
                         parquet_options=TableParseParquetOptions(
-                            coerce_int96_timestamp_unit=format_config.coerce_int96_timestamp_unit,
+                            coerce_int96_timestamp_unit=TimeUnit._from_pytimeunit(format_config.coerce_int96_timestamp_unit),
                         ),
                     )
                     for fp in filepaths
-Original file line number
+Diff line change
@@ Expand Up / @@ -224,6 +224,7 @@ class ParquetSourceConfig: @@
         """
         Configuration of a Parquet data source.
         """
         coerce_int96_timestamp_unit: PyTimeUnit | None
         def __init__(self, coerce_int96_timestamp_unit: PyTimeUnit | None = None): ...
@@ Expand Down @@