Fix issue with using .assign() with result from df.get() (#1463)

fix assign in conjunction with get

Author: Dr-Irv

pandas-stubs/_typing.pyi

@@ -1090,7 +1090,7 @@ TimeZones: TypeAlias = str | tzinfo | None | int
 IntoColumn: TypeAlias = (
     AnyArrayLike
     | Scalar
-    | Callable[[DataFrame], AnyArrayLike | Scalar | Sequence[Scalar] | range]
+    | Callable[[DataFrame], AnyArrayLike | Scalar | Sequence[Scalar] | range | None]
     | Sequence[Scalar]
     | range
     | None

tests/test_frame.py

@@ -415,6 +415,10 @@ def my_named_func_2(df: pd.DataFrame) -> pd.Series:
     check(assert_type(df.assign(c=my_named_func_1), pd.DataFrame), pd.DataFrame)
     check(assert_type(df.assign(c=my_named_func_2), pd.DataFrame), pd.DataFrame)
     check(assert_type(df.assign(c=None), pd.DataFrame), pd.DataFrame)
+    check(
+        assert_type(df.assign(foo=lambda df: df.get("abc", None)), pd.DataFrame),
+        pd.DataFrame,
+    )
 
 
 def test_types_sample() -> None: