gh-372 : Fixing Series.astype()

pandas-stubs/core/series.pyi

@@ -1,3 +1,9 @@
+from core.api import(
+    Int8Dtype as Int8Dtype,
+    Int16Dtype as Int16Dtype,
+    Int32Dtype as Int32Dtype,
+    Int64Dtype as Int64Dtype,
+)
 from collections.abc import (
     Callable,
     Hashable,
@@ -1025,12 +1031,26 @@ class Series(IndexOpsMixin, NDFrame, Generic[S1]):
         axis: SeriesAxisType | None = ...,
         ignore_index: _bool = ...,
     ) -> Series[S1]: ...
+    #def astype(
+     #   self,
+      #  dtype: S1 | _str | type[Scalar],
+       # copy: _bool = ...,
+        #errors: IgnoreRaise = ...,
+    #) -> Series: ...
+    @overload
     def astype(
         self,
-        dtype: S1 | _str | type[Scalar],
+        dtype: S1, _str,
         copy: _bool = ...,
-        errors: IgnoreRaise = ...,
-    ) -> Series: ...
+        errors: IgnoreRaise = ...
+        ) -> Series[str]: ...
+    @overload
+    def astype(
+        self,
+        dtype: S1, Int8Dtype, Int16Dtype, Int32Dtype, Int64Dtype,
+        copy: _bool = ...,
+        errors: IgnoreRaise = ...
+        ) -> Series[int]: ...
     def copy(self, deep: _bool = ...) -> Series[S1]: ...
     def infer_objects(self) -> Series[S1]: ...
     def convert_dtypes(

table.rst

@@ -0,0 +1,49 @@
+
+Table for astype
+
+
++-------------------+-------------------------+
+|  S1               |  Dtype                  |
++===================+=========================+
+|  int              | pd.Int8Dtype,           |
+|                   | pd.Int16Dtype,          |
+|                   | pd.Int32Dtype,          |
+|                   | pd.Int54Dtype,          |
+|                   | "int",int,np.int8,      |
+|                   |  np.int16,np.int32,     |
+|                   |  np.int64,np.uint8      |
+|                   |  np.uint16,np.uint32    |
+|                   |  np.uint64,np.intp      |
+|                   |  np.uintp,np.integer    |
++-------------------+-------------------------+
+|  str              | StringDtype,str,"str"   |
++-------------------+-------------------------+
+|  bytes            | np.byte,np.ubyte,bytes  |
+|                   | "bytes"                 | 
++-------------------+-------------------------+
+|  float            | pd.Float32Dtype,        |
+|                   | pd.Float64Dtype,        |
+|                   | float,"float",np.float16|
+|                   | np.float32,np.float64   |
+|                   | np.float96,np.float128  |
+|                   | np.floating             |
++-------------------+-------------------------+
+|  complex          | "complex",np.complex64, |
+|                   | np.complex128,          |
+|                   | np.complex192,          |
+|                   | np.complex256,          |
+|                   | complex                 |
++-------------------+-------------------------+
+|  bool             | pd.BooleanDtype, bool   |
+|                   | np.bool_                |
+|                   | "bool"                  |
++-------------------+-------------------------+
+| Timedelta         | np.timedelta64,         |
+|                   | Timedelta,              |
+|                   | "timedelta64[ns]"       |
+|                   |                         |
++-------------------+-------------------------+
+|  Timestamp        | Timestamp               |
+|                   | "datetime64[ns]"        |
++-------------------+-------------------------+
+