Skip to content

Commit ed30222

Browse files
committed
Add missing self arguments for __dlpack__ and __dlpack_device__
1 parent 8e83f0d commit ed30222

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

array_api_tests/function_stubs/array_object.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,13 +37,13 @@ def __bool__(x: array, /) -> bool:
3737
"""
3838
pass
3939

40-
def __dlpack__(*, stream: Optional[int] = None) -> PyCapsule:
40+
def __dlpack__(x: array, /, *, stream: Optional[int] = None) -> PyCapsule:
4141
"""
4242
Note: __dlpack__ is a method of the array object.
4343
"""
4444
pass
4545

46-
def __dlpack_device__() -> Tuple[IntEnum, int]:
46+
def __dlpack_device__(x: array, /) -> Tuple[IntEnum, int]:
4747
"""
4848
Note: __dlpack_device__ is a method of the array object.
4949
"""

0 commit comments

Comments
 (0)