Fix DeprecationWarning regarding throw() on pytests

Signed-off-by: Ziheng Deng <zihengd@nvidia.com>

Author: gunnersdeng

src/cuda/tile/_coroutine_util.py

@@ -20,7 +20,7 @@ def run_coroutine(awaitable: Awaitable):
             while stack:
                 top = stack[-1]
                 try:
-                    continuation = top.send(ret) if exc_info is None else top.throw(*exc_info)
+                    continuation = top.send(ret) if exc_info is None else top.throw(exc_info[1])
                 except StopIteration as s:
                     ret = s.value
                     exc_info = None

src/cuda/tile/_stub.py

@@ -1341,7 +1341,7 @@ def reduce(x, /, axis, func, identity, *, keepdims=False):
             being combined, while the rest correspond to the other.
         identity: a constant scalar or a tuple of constant scalars that specifies the identity
             element of the `func`.
-        keepdims (bool): True to remove the reduced axis, False to keep the axis of size 1.
+        keepdims (bool): True to keep the axis of size 1, False to remove the reduced axis.
             Default: False.
 
     Returns:

test/test_coroutine_util.py

@@ -4,6 +4,7 @@
 
 from cuda.tile._coroutine_util import resume_after, run_coroutine
 import pytest
+import traceback
 
 
 async def series(n):
@@ -60,3 +61,19 @@ async def two_calls():
 def test_return_values():
     res = run_coroutine(two_calls())
     assert res == (1 + 2 + 3, 1 + 2 + 3 + 4)
+
+
+async def raise_in_leaf():
+    raise ValueError("leaf")
+
+
+async def call_leaf():
+    await resume_after(raise_in_leaf())
+
+
+def test_traceback_preserved():
+    try:
+        run_coroutine(call_leaf())
+    except ValueError as e:
+        frame_names = [f.name for f in traceback.extract_tb(e.__traceback__)]
+        assert "raise_in_leaf" in frame_names