dycw-utilities 0.109.18__py3-none-any.whl → 0.109.19__py3-none-any.whl

{dycw_utilities-0.109.18.dist-info → dycw_utilities-0.109.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dycw-utilities
-Version: 0.109.18
+Version: 0.109.19
 Author-email: Derek Wan <d.wan@icloud.com>
 License-File: LICENSE
 Requires-Python: >=3.12

{dycw_utilities-0.109.18.dist-info → dycw_utilities-0.109.19.dist-info}/RECORD

@@ -1,4 +1,4 @@
-utilities/__init__.py,sha256=oi6A9do1IFvsNef8qyo7k6dU87Y8HcM7SGeQWrtR5w0,61
+utilities/__init__.py,sha256=NU7I4AynUyg4F6Vu4XytgM7aO4TDKxvPbBFKo638x30,61
 utilities/altair.py,sha256=Gpja-flOo-Db0PIPJLJsgzAlXWoKUjPU1qY-DQ829ek,9156
 utilities/astor.py,sha256=xuDUkjq0-b6fhtwjhbnebzbqQZAjMSHR1IIS5uOodVg,777
 utilities/asyncio.py,sha256=41oQUurWMvadFK5gFnaG21hMM0Vmfn2WS6OpC0R9mas,14757
@@ -46,8 +46,8 @@ utilities/pathlib.py,sha256=31WPMXdLIyXgYOMMl_HOI2wlo66MGSE-cgeelk-Lias,1410
 utilities/period.py,sha256=ikHXsWtDLr553cfH6p9mMaiCnIAP69B7q84ckWV3HaA,10884
 utilities/pickle.py,sha256=Bhvd7cZl-zQKQDFjUerqGuSKlHvnW1K2QXeU5UZibtg,657
 utilities/platform.py,sha256=NU7ycTvAXAG-fdYmDXaM1m4EOml2cGiaYwaUzfzSqyU,1767
-utilities/polars.py,sha256=tagkPZ-LlyyC9Mx-iCmLmCl3FlblKUVxWSXAaTh-G_M,53164
-utilities/polars_ols.py,sha256=Uc9V5kvlWZ5cU93lKZ-cfAKdVFFw81tqwLW9PxtUvMs,5618
+utilities/polars.py,sha256=UFrD7smyBROlIBaLD6tUFKGAPtwLfyHPHp6dJI9i-kQ,54961
+utilities/polars_ols.py,sha256=efhXf0gjrHUpQrvS6a7g8yJQJWf_ATKtJnqqF2inCOU,5680
 utilities/pqdm.py,sha256=foRytQybmOQ05pjt5LF7ANyzrIa--4ScDE3T2wd31a4,3118
 utilities/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 utilities/pydantic.py,sha256=f6qtR5mO2YMuyvNmbaEj5YeD9eGA4YYfb7Bjzh9jUs0,1845
@@ -86,7 +86,7 @@ utilities/warnings.py,sha256=yUgjnmkCRf6QhdyAXzl7u0qQFejhQG3PrjoSwxpbHrs,1819
 utilities/whenever.py,sha256=TjoTAJ1R27-rKXiXzdE4GzPidmYqm0W58XydDXp-QZM,17786
 utilities/zipfile.py,sha256=24lQc9ATcJxHXBPc_tBDiJk48pWyRrlxO2fIsFxU0A8,699
 utilities/zoneinfo.py,sha256=-DQz5a0Ikw9jfSZtL0BEQkXOMC9yGn_xiJYNCLMiqEc,1989
-dycw_utilities-0.109.18.dist-info/METADATA,sha256=QCV1ccFqiVe1zuFNo1eiBrkw5_X9YtyA409swR54OPM,13005
-dycw_utilities-0.109.18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dycw_utilities-0.109.18.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
-dycw_utilities-0.109.18.dist-info/RECORD,,
+dycw_utilities-0.109.19.dist-info/METADATA,sha256=nYNr6kZIYFI-2yrn7iwBAs_Axo9Zqi5-bKBZpNBc1Q0,13005
+dycw_utilities-0.109.19.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dycw_utilities-0.109.19.dist-info/licenses/LICENSE,sha256=gppZp16M6nSVpBbUBrNL6JuYfvKwZiKgV7XoKKsHzqo,1066
+dycw_utilities-0.109.19.dist-info/RECORD,,

utilities/__init__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-__version__ = "0.109.18"
+__version__ = "0.109.19"

utilities/polars.py

@@ -115,6 +115,7 @@ if TYPE_CHECKING:
         JoinStrategy,  # pyright: ignore[reportPrivateImportUsage]
         JoinValidation,  # pyright: ignore[reportPrivateImportUsage]
         PolarsDataType,  # pyright: ignore[reportPrivateImportUsage]
+        RollingInterpolationMethod,  # pyright: ignore[reportPrivateImportUsage]
         SchemaDict,  # pyright: ignore[reportPrivateImportUsage]
         TimeUnit,  # pyright: ignore[reportPrivateImportUsage]
     )
@@ -680,6 +681,73 @@ def _cross_or_touch(
 ##
 
 
+@overload
+def cross_rolling_quantile(
+    expr: ExprLike,
+    up_or_down: Literal["up", "down"],
+    quantile: float,
+    /,
+    *,
+    interpolation: RollingInterpolationMethod = "nearest",
+    window_size: int = 2,
+    weights: list[float] | None = None,
+    min_samples: int | None = None,
+    center: bool = False,
+) -> Expr: ...
+@overload
+def cross_rolling_quantile(
+    expr: Series,
+    up_or_down: Literal["up", "down"],
+    quantile: float,
+    /,
+    *,
+    interpolation: RollingInterpolationMethod = "nearest",
+    window_size: int = 2,
+    weights: list[float] | None = None,
+    min_samples: int | None = None,
+    center: bool = False,
+) -> Series: ...
+@overload
+def cross_rolling_quantile(
+    expr: IntoExprColumn,
+    up_or_down: Literal["up", "down"],
+    quantile: float,
+    /,
+    *,
+    interpolation: RollingInterpolationMethod = "nearest",
+    window_size: int = 2,
+    weights: list[float] | None = None,
+    min_samples: int | None = None,
+    center: bool = False,
+) -> Expr | Series: ...
+def cross_rolling_quantile(
+    expr: IntoExprColumn,
+    up_or_down: Literal["up", "down"],
+    quantile: float,
+    /,
+    *,
+    interpolation: RollingInterpolationMethod = "nearest",
+    window_size: int = 2,
+    weights: list[float] | None = None,
+    min_samples: int | None = None,
+    center: bool = False,
+) -> Expr | Series:
+    """Compute when a column crosses its rolling quantile."""
+    expr = ensure_expr_or_series(expr)
+    rolling = expr.rolling_quantile(
+        quantile,
+        interpolation=interpolation,
+        window_size=window_size,
+        weights=weights,
+        min_samples=min_samples,
+        center=center,
+    )
+    return cross(expr, up_or_down, rolling)
+
+
+##
+
+
 def dataclass_to_dataframe(
     objs: MaybeIterable[Dataclass],
     /,

utilities/polars_ols.py

@@ -10,7 +10,9 @@ from utilities.functions import is_sequence_of
 from utilities.polars import concat_series, ensure_expr_or_series
 
 if TYPE_CHECKING:
-    from polars._typing import IntoExprColumn
+    from polars._typing import (
+        IntoExprColumn,  # pyright: ignore[reportPrivateImportUsage]
+    )
     from polars_ols import NullPolicy
 
     from utilities.polars import ExprLike