fix: assign Sequences (#1260)

fix: assign empty and tuple

Author: cmp0xff

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
 )

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]})