GH1173 PR feedback

Author: loicdiridollou

pandas-stubs/core/frame.pyi

@@ -706,7 +706,14 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
         level: int = ...,
         target: object | None = ...,
         inplace: Literal[True],
-        **kwargs: Any,  # TODO: make more precise https://github.com/pandas-dev/pandas-stubs/issues/1173
+    ) -> None: ...
+    @overload
+    def query(
+        self,
+        expr: _str,
+        *,
+        inplace: Literal[True],
+        **kwargs: Any,
     ) -> None: ...
     @overload
     def query(
@@ -721,7 +728,14 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
         resolvers: list[Mapping] | None = ...,
         level: int = ...,
         target: object | None = ...,
-        **kwargs: Any,  # TODO: make more precise https://github.com/pandas-dev/pandas-stubs/issues/1173
+    ) -> Self: ...
+    @overload
+    def query(
+        self,
+        expr: _str,
+        *,
+        inplace: Literal[False] = ...,
+        **kwargs: Any,
     ) -> Self: ...
     @overload
     def eval(self, expr: _str, *, inplace: Literal[True], **kwargs: Any) -> None: ...

tests/test_frame.py

@@ -529,7 +529,10 @@ def test_types_query_kwargs() -> None:
         pd.DataFrame,
     )
     kwargs = {"parser": "pandas", "engine": "numexpr"}
-    check(assert_type(df.query("col1 > col2", inplace=False, **kwargs), pd.DataFrame), pd.DataFrame)
+    check(
+        assert_type(df.query("col1 > col2", inplace=False, **kwargs), pd.DataFrame),
+        pd.DataFrame,
+    )
 
     check(assert_type(df.query("col1 % col2 == 0", inplace=True), None), type(None))