add overload defs for level

Author: yadav-sachin

pandas/core/frame.py

@@ -8642,6 +8642,28 @@ def aggregate(self, func=None, axis: Axis = 0, *args, **kwargs):
 
     agg = aggregate
 
+    @overload
+    def any(
+        self: DataFrame,
+        axis: Axis = ...,
+        bool_only: bool | None = None,
+        skipna: bool = ...,
+        level: None = ...,
+        **kwargs,
+    ) -> Series:
+        ...
+
+    @overload
+    def any(
+        self: DataFrame,
+        axis: Axis = ...,
+        bool_only: bool | None = None,
+        skipna: bool = ...,
+        level: Level = ...,
+        **kwargs,
+    ) -> DataFrame:
+        ...
+
     @doc(NDFrame.any, **_shared_doc_kwargs)
     def any(
         self: DataFrame,
@@ -8650,7 +8672,7 @@ def any(
         skipna: bool = True,
         level: Level | None = None,
         **kwargs,
-    ) -> Series:
+    ) -> Series | DataFrame:
         return super().any(
             axis=axis, bool_only=bool_only, skipna=skipna, level=level, **kwargs
         )

pandas/core/series.py

@@ -4258,6 +4258,28 @@ def aggregate(self, func=None, axis=0, *args, **kwargs):
 
     agg = aggregate
 
+    @overload
+    def any(
+        self: Series,
+        axis: Axis = ...,
+        bool_only: bool | None = None,
+        skipna: bool = ...,
+        level: None = ...,
+        **kwargs,
+    ) -> bool:
+        ...
+
+    @overload
+    def any(
+        self: Series,
+        axis: Axis = ...,
+        bool_only: bool | None = None,
+        skipna: bool = ...,
+        level: Level = ...,
+        **kwargs,
+    ) -> Series:
+        ...
+
     @doc(NDFrame.any, **_shared_doc_kwargs)
     def any(
         self: Series,
@@ -4266,7 +4288,7 @@ def any(
         skipna: bool = True,
         level: Level | None = None,
         **kwargs,
-    ) -> bool:
+    ) -> bool | Series:
         return super().any(
             axis=axis, bool_only=bool_only, skipna=skipna, level=level, **kwargs
         )