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,