type window/* (#1564)

type window

Author: MarcoGorelli

pandas-stubs/core/window/ewm.pyi

@@ -1,5 +1,3 @@
-from typing import Any
-
 from pandas import (
     DataFrame,
     Series,
@@ -47,29 +45,3 @@ class ExponentialMovingWindow(BaseWindow[NDFrameT]):
 class ExponentialMovingWindowGroupby(
     BaseWindowGroupby[NDFrameT], ExponentialMovingWindow[NDFrameT]
 ): ...
-
-class OnlineExponentialMovingWindow(ExponentialMovingWindow[NDFrameT]):
-    def reset(self) -> None: ...
-    def aggregate(self, func, *args: Any, **kwargs: Any): ...
-    def std(self, bias: bool = False, *args: Any, **kwargs: Any): ...
-    def corr(
-        self,
-        other: DataFrame | Series | None = None,
-        pairwise: bool | None = None,
-        numeric_only: bool = False,
-    ): ...
-    def cov(
-        self,
-        other: DataFrame | Series | None = None,
-        pairwise: bool | None = None,
-        bias: bool = False,
-        numeric_only: bool = False,
-    ): ...
-    def var(self, bias: bool = False, numeric_only: bool = False): ...
-    def mean(
-        self,
-        *args: Any,
-        update: NDFrameT | None = ...,
-        update_times: None = None,
-        **kwargs: Any,
-    ) -> NDFrameT: ...

pandas-stubs/core/window/rolling.pyi

@@ -1,6 +1,8 @@
 from collections.abc import (
     Callable,
+    Hashable,
     Iterator,
+    Sequence,
 )
 import datetime as dt
 from typing import (
@@ -42,7 +44,7 @@ class BaseWindow(SelectionMixin[NDFrameT]):
     win_type: str | None
     axis: AxisInt
     method: CalculationMethod
-    def __getitem__(self, key) -> Self: ...
+    def __getitem__(self, key: Hashable | Sequence[Hashable]) -> Self: ...
     def __getattr__(self, attr: str) -> Self: ...
     def __iter__(self) -> Iterator[NDFrameT]: ...
     @overload

pyproject.toml

@@ -209,10 +209,6 @@ ignore = [
   # TODO: remove when computations are fully typed
   "ANN001", "ANN201", "ANN204", "ANN206",
 ]
-"*window*" = [
-  # TODO: remove when window is fully typed
-  "ANN001", "ANN201", "ANN204", "ANN206",
-]
 "*generic.pyi" = [
   # TODO: remove when generic.pyi is fully typed
   "ANN001", "ANN201", "ANN204", "ANN206",