dycw-utilities 0.138.10__py3-none-any.whl → 0.138.12__py3-none-any.whl

{dycw_utilities-0.138.10.dist-info → dycw_utilities-0.138.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.138.10
+Version: 0.138.12
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12
@@ -12,7 +12,7 @@ Provides-Extra: logging
 Requires-Dist: coloredlogs<15.1,>=15.0.1; extra == 'logging'
 Provides-Extra: test
 Requires-Dist: dycw-pytest-only<2.2,>=2.1.1; extra == 'test'
-Requires-Dist: hypothesis<6.136,>=6.135.14; extra == 'test'
+Requires-Dist: hypothesis<6.136,>=6.135.16; extra == 'test'
 Requires-Dist: pudb<2025.2,>=2025.1; extra == 'test'
 Requires-Dist: pytest-asyncio<1.1,>=1.0.0; extra == 'test'
 Requires-Dist: pytest-cov<6.2,>=6.1.1; extra == 'test'

{dycw_utilities-0.138.10.dist-info → dycw_utilities-0.138.12.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=KXSG109KauV-mzYqWH_b1979j4n9biCSq0kQFI5m5Z4,61
+utilities/__init__.py,sha256=8Og3Xw4iUhz4bYD_NYUCU8ER2AMFKi6vrYO5Sk_f95M,61
 utilities/aiolimiter.py,sha256=mD0wEiqMgwpty4XTbawFpnkkmJS6R4JRsVXFUaoitSU,628
 utilities/altair.py,sha256=HeZBVUocjkrTNwwKrClppsIqgNFF-ykv05HfZSoHYno,9104
 utilities/asyncio.py,sha256=dcGeKQzjLBXxKzZkVIk5oZsFXEcynVbRB9iNB5XEDZk,38526
@@ -46,7 +46,7 @@ utilities/pathlib.py,sha256=5p9MI8m7iUFksNVGiLavN4MZ3cj1CpBiLBlgaCR0LCs,8334
 utilities/period.py,sha256=6jEff_qAiE7xdFaQ1DnKgNf10D2wHhzt7hQXCBoKlgc,6842
 utilities/pickle.py,sha256=MBT2xZCsv0pH868IXLGKnlcqNx2IRVKYNpRcqiQQqxw,653
 utilities/platform.py,sha256=5uCKRf_ij7ukJDcbnNfhY2ay9fbrpiNLRO1t2QvcwqQ,2825
-utilities/polars.py,sha256=xCeB-pLkezOoQJvgxTdz2rNlaehdhek-HmDztXWn2j0,63266
+utilities/polars.py,sha256=Q7eGBZ-P3tg2HXVk-rF7yceodeuPoj-sfPbYWawqyRM,63886
 utilities/polars_ols.py,sha256=Uc9V5kvlWZ5cU93lKZ-cfAKdVFFw81tqwLW9PxtUvMs,5618
 utilities/pottery.py,sha256=RN3XwOEsVAPXvEfsRPmn3ZSKgTzK_c182PNrtksq-bg,3429
 utilities/pqdm.py,sha256=BTsYPtbKQWwX-iXF4qCkfPG7DPxIB54J989n83bXrIo,3092
@@ -85,11 +85,11 @@ utilities/tzlocal.py,sha256=KyCXEgCTjqGFx-389JdTuhMRUaT06U1RCMdWoED-qro,728
 utilities/uuid.py,sha256=32p7DGHGM2Btx6PcBvCZvERSWbpupMXqx6FppPoSoTU,612
 utilities/version.py,sha256=ufhJMmI6KPs1-3wBI71aj5wCukd3sP_m11usLe88DNA,5117
 utilities/warnings.py,sha256=un1LvHv70PU-LLv8RxPVmugTzDJkkGXRMZTE2-fTQHw,1771
-utilities/whenever.py,sha256=R5d9UCNCdAOyjwLUmfH2Vn8Ykee8OHQi2skRTFfbZMM,20492
+utilities/whenever.py,sha256=sUAOQ2sJTRGC1kacGTnMbfQdkypuSiqhTZWZVo21OT8,21545
 utilities/zipfile.py,sha256=24lQc9ATcJxHXBPc_tBDiJk48pWyRrlxO2fIsFxU0A8,699
 utilities/zoneinfo.py,sha256=oEH-nL3t4h9uawyZqWDtNtDAl6M-CLpLYGI_nI6DulM,1971
-dycw_utilities-0.138.10.dist-info/METADATA,sha256=5_CxObrlGLTgwHzV1c8558AlnuS2yZx8skO-M6OisCM,1639
-dycw_utilities-0.138.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.138.10.dist-info/entry_points.txt,sha256=uLj5QWWVXv8tnMaRX3ZGYpt7w1xzLWU6LxbFhELEpkc,68
-dycw_utilities-0.138.10.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.138.10.dist-info/RECORD,,
+dycw_utilities-0.138.12.dist-info/METADATA,sha256=1DfnY3KS8qcT0z9ClK0-Ew_aXn0d5-SIsApG2pminew,1639
+dycw_utilities-0.138.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.138.12.dist-info/entry_points.txt,sha256=uLj5QWWVXv8tnMaRX3ZGYpt7w1xzLWU6LxbFhELEpkc,68
+dycw_utilities-0.138.12.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.138.12.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.138.10"
+__version__ = "0.138.12"

utilities/polars.py

@@ -1833,6 +1833,26 @@ def normal(
 ##
 
 
+@overload
+def order_of_magnitude(column: ExprLike, /, *, round_: bool = False) -> Expr: ...
+@overload
+def order_of_magnitude(column: Series, /, *, round_: bool = False) -> Series: ...
+@overload
+def order_of_magnitude(
+    column: IntoExprColumn, /, *, round_: bool = False
+) -> Expr | Series: ...
+def order_of_magnitude(
+    column: IntoExprColumn, /, *, round_: bool = False
+) -> Expr | Series:
+    """Compute the order of magnitude of a column."""
+    column = ensure_expr_or_series(column)
+    result = column.abs().log10()
+    return result.round().cast(Int64) if round_ else result
+
+
+##
+
+
 def reify_exprs(
     *exprs: IntoExprColumn, **named_exprs: IntoExprColumn
 ) -> Expr | Series | DataFrame:
@@ -2184,6 +2204,7 @@ __all__ = [
     "nan_sum_agg",
     "nan_sum_cols",
     "normal",
+    "order_of_magnitude",
     "replace_time_zone",
     "set_first_row_as_columns",
     "struct_dtype",

utilities/whenever.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import datetime as dt
 from collections.abc import Callable, Iterable, Mapping
 from dataclasses import dataclass
 from functools import cache
@@ -553,9 +554,10 @@ def to_local_plain(date_time: ZonedDateTime, /) -> PlainDateTime:
 
 def to_nanos(delta: DateTimeDelta, /) -> int:
     """Compute the number of nanoseconds in a date-time delta."""
-    months, days, _, _ = delta.in_months_days_secs_nanos()
-    if months != 0:
-        raise ToNanosError(months=months)
+    try:
+        days = to_days(delta.date_part())
+    except ToDaysError as error:
+        raise ToNanosError(months=error.months) from None
     return 24 * 60 * 60 * int(1e9) * days + delta.time_part().in_nanoseconds()
 
 
@@ -571,6 +573,37 @@ class ToNanosError(Exception):
 ##
 
 
+def to_py_time_delta(delta: DateDelta | TimeDelta | DateTimeDelta, /) -> dt.timedelta:
+    """Try convert a DateDelta to a standard library timedelta."""
+    match delta:
+        case DateDelta():
+            return dt.timedelta(days=to_days(delta))
+        case TimeDelta():
+            nanos = delta.in_nanoseconds()
+            micros, remainder = divmod(nanos, 1000)
+            if remainder != 0:
+                raise ToPyTimeDeltaError(nanoseconds=remainder)
+            return dt.timedelta(microseconds=micros)
+        case DateTimeDelta():
+            return to_py_time_delta(delta.date_part()) + to_py_time_delta(
+                delta.time_part()
+            )
+        case _ as never:
+            assert_never(never)
+
+
+@dataclass(kw_only=True, slots=True)
+class ToPyTimeDeltaError(Exception):
+    nanoseconds: int
+
+    @override
+    def __str__(self) -> str:
+        return f"Time delta must not contain nanoseconds; got {self.nanoseconds}"
+
+
+##
+
+
 def to_time_delta(nanos: int, /) -> TimeDelta:
     """Construct a time delta."""
     components = _to_time_delta_components(nanos)
@@ -769,6 +802,7 @@ __all__ = [
     "MinMaxDateError",
     "ToDaysError",
     "ToNanosError",
+    "ToPyTimeDeltaError",
     "WheneverLogRecord",
     "datetime_utc",
     "format_compact",
@@ -786,6 +820,7 @@ __all__ = [
     "to_days",
     "to_local_plain",
     "to_nanos",
+    "to_py_time_delta",
     "to_zoned_date_time",
     "two_digit_year_month",
 ]