added any to list of args for astype

Author: ramvikrams

pandas-stubs/_typing.pyi

@@ -76,6 +76,7 @@ class FulldatetimeDict(YearMonthDayDict, total=False):
 # dtypes
 NpDtype: TypeAlias = str | np.dtype[np.generic] | type[str | complex | bool | object]
 Dtype: TypeAlias = ExtensionDtype | NpDtype
+DtypeArg: TypeAlias = Dtype | dict[Any, Dtype]
 AstypeArg: TypeAlias = (
     BooleanDtypeArg
     | IntDtypeArg
@@ -89,7 +90,6 @@ AstypeArg: TypeAlias = (
     | ExtensionDtype
 )
 # DtypeArg specifies all allowable dtypes in a functions its dtype argument
-DtypeArg: TypeAlias = Dtype | dict[Any, Dtype]
 DtypeObj: TypeAlias = np.dtype[np.generic] | ExtensionDtype
 
 # filenames and file-like-objects

pandas-stubs/core/frame.pyi

@@ -1441,8 +1441,7 @@ class DataFrame(NDFrame, OpsMixin):
     ) -> DataFrame: ...
     def astype(
         self,
-        dtype: AstypeArg | Mapping[Hashable, AstypeArg] | Series,
-        # dtype: AstypeArg | dict[Any, AstypeArg],
+        dtype: AstypeArg | Mapping[Hashable, AstypeArg] | Series | Any,
         copy: _bool = ...,
         errors: IgnoreRaise = ...,
     ) -> DataFrame: ...

tests/test_frame.py

@@ -2296,6 +2296,9 @@ def test_axes_as_tuple() -> None:
 def test_astype_dict() -> None:
     # GH 447
     df = pd.DataFrame({"a": [1, 2, 3], 43: [4, 5, 6]})
+    columns_types = {"a": "int", 43: "float"}
+    de = df.astype(columns_types)
+    check(assert_type(de, pd.DataFrame), pd.DataFrame)
     check(assert_type(df.astype({"a": "int", 43: "float"}), pd.DataFrame), pd.DataFrame)