Stub of __cuda_stream__ special method declare tuple[int, int] as return type

This is to indicate that special method always returns a pair of integers
This commit is contained in:
Oleksandr Pavlyk
2025-08-04 10:11:33 -05:00
parent a5e0a48f80
commit 73e18419b2

View File

@@ -35,7 +35,7 @@ class CudaStream:
----
The class is not directly constructible.
"""
def __cuda_stream__(self) -> tuple[int]:
def __cuda_stream__(self) -> tuple[int, int]:
"""
Special method implement CUDA stream protocol
from `cuda.core`. Returns a pair of integers: