kumoai 2.13.0.dev202511211730__py3-none-any.whl → 2.15.0.dev202601131732__py3-none-any.whl

kumoai/experimental/rfm/base/utils.py

@@ -0,0 +1,36 @@
+import warnings
+
+import pandas as pd
+import pyarrow as pa
+
+
+def is_datetime(ser: pd.Series) -> bool:
+    r"""Check whether a :class:`pandas.Series` holds datetime values."""
+    if isinstance(ser.dtype, pd.ArrowDtype):
+        dtype = ser.dtype.pyarrow_dtype
+        return (pa.types.is_timestamp(dtype) or pa.types.is_date(dtype)
+                or pa.types.is_time(dtype))
+
+    return pd.api.types.is_datetime64_any_dtype(ser)
+
+
+def to_datetime(ser: pd.Series) -> pd.Series:
+    """Converts a :class:`pandas.Series` to ``datetime64[ns]`` format."""
+    if isinstance(ser.dtype, pd.ArrowDtype):
+        ser = pd.Series(ser.to_numpy(), index=ser.index, name=ser.name)
+
+    if not pd.api.types.is_datetime64_any_dtype(ser):
+        with warnings.catch_warnings():
+            warnings.filterwarnings(
+                'ignore',
+                message='Could not infer format',
+            )
+            ser = pd.to_datetime(ser, errors='coerce')
+
+    if isinstance(ser.dtype, pd.DatetimeTZDtype):
+        ser = ser.dt.tz_localize(None)
+
+    if ser.dtype != 'datetime64[ns]':
+        ser = ser.astype('datetime64[ns]')
+
+    return ser