Update a function signature from the array API spec

Author: asmeurer

numpy/_array_api/_array_object.py

@@ -22,7 +22,7 @@
 
 from typing import TYPE_CHECKING
 if TYPE_CHECKING:
-    from ._types import Optional, PyCapsule, Tuple, Union, array
+    from ._types import Any, Optional, PyCapsule, Tuple, Union, array
 
 import numpy as np
 
@@ -186,7 +186,7 @@ def __bool__(self: array, /) -> bool:
         res = self._array.__bool__()
         return res
 
-    def __dlpack__(self: array, /, *, stream: Optional[int] = None) -> PyCapsule:
+    def __dlpack__(self: array, /, *, stream: Optional[Union[int, Any]] = None) -> PyCapsule:
         """
         Performs the operation __dlpack__.
         """

numpy/_array_api/_types.py

@@ -6,10 +6,11 @@
 valid for inputs that match the given type annotations.
 """
 
-__all__ = ['List', 'Literal', 'Optional', 'Tuple', 'Union', 'array', 'device',
-           'dtype', 'SupportsDLPack', 'SupportsBufferProtocol', 'PyCapsule']
+__all__ = ['Any', 'List', 'Literal', 'Optional', 'Tuple', 'Union', 'array',
+           'device', 'dtype', 'SupportsDLPack', 'SupportsBufferProtocol',
+           'PyCapsule']
 
-from typing import List, Literal, Optional, Tuple, Union, TypeVar
+from typing import Any, List, Literal, Optional, Tuple, Union, TypeVar
 
 from . import (ndarray, int8, int16, int32, int64, uint8, uint16, uint32,
                uint64, float32, float64)