From c8d241a861c257dbb02ae5a7e52871cdeaad1b32 Mon Sep 17 00:00:00 2001 From: cmp0xff Date: Mon, 23 Jun 2025 15:14:34 +0200 Subject: [PATCH] fix: assign empty and tuple --- pandas-stubs/_typing.pyi | 4 ++-- tests/test_frame.py | 9 ++++++++- 2 files changed, 10 insertions(+), 3 deletions(-) diff --git a/pandas-stubs/_typing.pyi b/pandas-stubs/_typing.pyi index 20c6849e8..5cef91fde 100644 --- a/pandas-stubs/_typing.pyi +++ b/pandas-stubs/_typing.pyi @@ -998,8 +998,8 @@ TimeZones: TypeAlias = str | tzinfo | None | int IntoColumn: TypeAlias = ( AnyArrayLike | Scalar - | Callable[[DataFrame], AnyArrayLike | Scalar | list[Scalar] | range] - | list[Scalar] + | Callable[[DataFrame], AnyArrayLike | Scalar | Sequence[Scalar] | range] + | Sequence[Scalar] | range | None ) diff --git a/tests/test_frame.py b/tests/test_frame.py index d58bd7f89..ffd85d609 100644 --- a/tests/test_frame.py +++ b/tests/test_frame.py @@ -317,7 +317,11 @@ def test_types_assign() -> None: df = pd.DataFrame({"a": [1, 2, 3]}) check( assert_type( - df.assign(b=lambda df: range(len(df)), c=lambda _: [10, 20, 30]), + df.assign( + b=lambda df: range(len(df)), + c=lambda _: [10, 20, 30], + d=lambda _: (10, 20, 30), + ), pd.DataFrame, ), pd.DataFrame, @@ -327,6 +331,9 @@ def test_types_assign() -> None: pd.DataFrame, ) + df = pd.DataFrame() + check(assert_type(df.assign(a=[], b=()), pd.DataFrame), pd.DataFrame) + def test_assign() -> None: df = pd.DataFrame({"a": [1, 2, 3], 1: [4, 5, 6]})