cont

Signed-off-by: nstarman <[email protected]>

Author: nstarman

src/array_api_stubs/_draft/array_object.py

@@ -1,16 +1,16 @@
 from __future__ import annotations
 
 from typing import TYPE_CHECKING, Protocol, TypeVar
+from .data_types import DType
+from ._types import device as Device
 
 if TYPE_CHECKING:
     from ._types import (
-        device as Device,
         Any,
         PyCapsule,
         Enum,
         ellipsis,
     )
-    from .data_types import DType
 
 array = TypeVar("array", bound="Array")
 # NOTE: when working with py3.11+ this can be ``typing.Self``.
@@ -376,7 +376,7 @@ def __dlpack_device__(self, /) -> tuple[Enum, int]:
 
         Returns
         -------
-        device: Tuple[Enum, int]
+        device: Tuple[:class:`~enum.Enum`, int]
             a tuple ``(device_type, device_id)`` in DLPack format. Valid device type enum members are:
 
             ::
@@ -1088,7 +1088,7 @@ def to_device(
         device: device
             a ``device`` object (see :ref:`device-support`).
         stream: Optional[Union[int, Any]]
-            stream object to use during copy. In addition to the types supported in :meth:`array.__dlpack__`, implementations may choose to support any library-specific stream object with the caveat that any code using such an object would not be portable.
+            stream object to use during copy. In addition to the types supported in :meth:`Array.__dlpack__`, implementations may choose to support any library-specific stream object with the caveat that any code using such an object would not be portable.
 
         Returns
         -------