From 73e18419b2f6f458c2867ee3607b4931288f08e6 Mon Sep 17 00:00:00 2001
From: Oleksandr Pavlyk <21087696+oleksandr-pavlyk@users.noreply.github.com>
Date: Mon, 4 Aug 2025 10:11:33 -0500
Subject: [PATCH] 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
---
 python/cuda/nvbench/__init__.pyi | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/cuda/nvbench/__init__.pyi b/python/cuda/nvbench/__init__.pyi
index 80a6dee..6fea984 100644
--- a/python/cuda/nvbench/__init__.pyi
+++ b/python/cuda/nvbench/__init__.pyi
@@ -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: