fix: Type hints for window_size incorrectly included timedelta in som…

…e rolling functions (#19827)
pola-rs · Nov 17, 2024 · 01be57f · 01be57f
1 parent 50c1135
commit 01be57f
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/py-polars/polars/expr/expr.py b/py-polars/polars/expr/expr.py
@@ -7657,7 +7657,7 @@ def rolling_mean(
     @unstable()
     def rolling_sum(
         self,
-        window_size: int | timedelta,
+        window_size: int,
         weights: list[float] | None = None,
         *,
         min_periods: int | None = None,
@@ -7768,7 +7768,7 @@ def rolling_sum(
     @unstable()
     def rolling_std(
         self,
-        window_size: int | timedelta,
+        window_size: int,
         weights: list[float] | None = None,
         *,
         min_periods: int | None = None,
@@ -7883,7 +7883,7 @@ def rolling_std(
     @unstable()
     def rolling_var(
         self,
-        window_size: int | timedelta,
+        window_size: int,
         weights: list[float] | None = None,
         *,
         min_periods: int | None = None,
@@ -7998,7 +7998,7 @@ def rolling_var(
     @unstable()
     def rolling_median(
         self,
-        window_size: int | timedelta,
+        window_size: int,
         weights: list[float] | None = None,
         *,
         min_periods: int | None = None,
@@ -8111,7 +8111,7 @@ def rolling_quantile(
         self,
         quantile: float,
         interpolation: RollingInterpolationMethod = "nearest",
-        window_size: int | timedelta = 2,
+        window_size: int = 2,
         weights: list[float] | None = None,
         *,
         min_periods: int | None = None,