Skip to content

Commit 02b0d7f

Browse files
GH1053 @ for DataFrame, eval in place for DataFrame, test migrations (#1054)
* GH1053 @ for DataFrame, eval in place for DataFrame, test migrations * Formatting and spelling * GH1053 PR Feedback * GH1053 Formatting
1 parent c5eecf8 commit 02b0d7f

File tree

2 files changed

+197
-103
lines changed

2 files changed

+197
-103
lines changed

pandas-stubs/core/frame.pyi

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -281,7 +281,12 @@ class DataFrame(NDFrame, OpsMixin):
281281
def dot(self, other: DataFrame | ArrayLike) -> DataFrame: ...
282282
@overload
283283
def dot(self, other: Series) -> Series: ...
284-
def __matmul__(self, other): ...
284+
@overload
285+
def __matmul__(self, other: DataFrame) -> DataFrame: ...
286+
@overload
287+
def __matmul__(self, other: Series) -> Series: ...
288+
@overload
289+
def __matmul__(self, other: np.ndarray) -> DataFrame: ...
285290
def __rmatmul__(self, other): ...
286291
@overload
287292
@classmethod
@@ -620,7 +625,12 @@ class DataFrame(NDFrame, OpsMixin):
620625
def query(
621626
self, expr: _str, *, inplace: Literal[False] = ..., **kwargs
622627
) -> DataFrame: ...
623-
def eval(self, expr: _str, *, inplace: _bool = ..., **kwargs): ...
628+
@overload
629+
def eval(self, expr: _str, *, inplace: Literal[True], **kwargs) -> None: ...
630+
@overload
631+
def eval(
632+
self, expr: _str, *, inplace: Literal[False] = ..., **kwargs
633+
) -> Scalar | np.ndarray | DataFrame | Series: ...
624634
AstypeArgExt: TypeAlias = (
625635
AstypeArg
626636
| Literal[

0 commit comments

Comments
 (0)