From 7b4fa1d71e1329b39aaac868fb6eaa68d71217e3 Mon Sep 17 00:00:00 2001 From: cmp0xff Date: Wed, 4 Jun 2025 09:06:31 +0200 Subject: [PATCH] refactor: #1238 simplify typing --- pandas-stubs/_typing.pyi | 2 +- pandas-stubs/core/frame.pyi | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pandas-stubs/_typing.pyi b/pandas-stubs/_typing.pyi index 28ee2fff..f4747598 100644 --- a/pandas-stubs/_typing.pyi +++ b/pandas-stubs/_typing.pyi @@ -1011,7 +1011,7 @@ TimeZones: TypeAlias = str | tzinfo | None | int # Evaluates to a DataFrame column in DataFrame.assign context. IntoColumn: TypeAlias = ( - AnyArrayLike | Scalar | Callable[[DataFrame], AnyArrayLike | Scalar] + AnyArrayLike | Scalar | Callable[[DataFrame], AnyArrayLike | Scalar] | None ) DatetimeLike: TypeAlias = datetime.datetime | np.datetime64 | Timestamp diff --git a/pandas-stubs/core/frame.pyi b/pandas-stubs/core/frame.pyi index c0319d21..f872ad2f 100644 --- a/pandas-stubs/core/frame.pyi +++ b/pandas-stubs/core/frame.pyi @@ -794,7 +794,7 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack): value: Scalar | ListLikeU | None, allow_duplicates: _bool = ..., ) -> None: ... - def assign(self, **kwargs: IntoColumn | None) -> Self: ... + def assign(self, **kwargs: IntoColumn) -> Self: ... def align( self, other: NDFrameT,