Further _from_sequence (#899)

dask · Oct 26, 2023 · 1f90f1a · 1f90f1a
1 parent 89acf38
commit 1f90f1a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fastparquet/dataframe.py b/fastparquet/dataframe.py
@@ -199,7 +199,7 @@ def set_cats(values, i=i, col=col, **kwargs):
         elif isinstance(bvalues.dtype, DatetimeTZDtype):
             dt = "M8[ns]" if PANDAS_VERSION.major < 2 else f'M8[{bvalues.dtype.unit}]'
             values = np.zeros(shape=shape, dtype=dt)
-            values = type(bvalues)._from_sequence(values, copy=False, dtype=bvalues.dtype)
+            values = type(bvalues)._from_sequence(values.view("int64"), copy=False, dtype=bvalues.dtype)
         else:
             if not isinstance(bvalues, np.ndarray):
                 # e.g. DatetimeLikeBlock backed by DatetimeArray/TimedeltaArray
@@ -210,7 +210,7 @@ def set_cats(values, i=i, col=col, **kwargs):
                 elif bvalues.dtype.kind == "M":
                     dt = "M8[ns]" if PANDAS_VERSION.major < 2 else bvalues.dtype
                     values = np.zeros(shape=shape, dtype=dt)
-                    values = type(bvalues)._from_sequence(values, copy=False)
+                    values = type(bvalues)._from_sequence(values.view("int64"), copy=False, dtype=bvalues.dtype)
                 elif str(bvalues.dtype)[0] in {"I", "U"} or str(bvalues.dtype) == "boolean":
                     arr_type = bvalues.dtype.construct_array_type()
                     values = arr_type(