add device kwarg to astype API

Author: steff456

src/array_api_stubs/_draft/data_type_functions.py

@@ -1,7 +1,19 @@
-from ._types import Union, Tuple, array, dtype, finfo_object, iinfo_object
-
-
-def astype(x: array, dtype: dtype, /, *, copy: bool = True) -> array:
+from ._types import (
+    Union,
+    Tuple,
+    array,
+    dtype,
+    finfo_object,
+    iinfo_object,
+    device,
+    Optional
+)
+
+
+def astype(
+    x: array, dtype: dtype, /, *, copy: bool = True,
+    device: Optional[device] = None
+) -> array:
     """
     Copies an array to a specified data type irrespective of :ref:`type-promotion` rules.
 
@@ -31,6 +43,8 @@ def astype(x: array, dtype: dtype, /, *, copy: bool = True) -> array:
         desired data type.
     copy: bool
         specifies whether to copy an array when the specified ``dtype`` matches the data type of the input array ``x``. If ``True``, a newly allocated array must always be returned. If ``False`` and the specified ``dtype`` matches the data type of the input array, the input array must be returned; otherwise, a newly allocated array must be returned. Default: ``True``.
+    device: Optional[device]
+        device on which to place the created array. Default: ``None``.
 
     Returns
     -------