From d8b0acc8d4fd3f893fee4b3800b5a4e561f97867 Mon Sep 17 00:00:00 2001
From: Oleksandr Pavlyk <21087696+oleksandr-pavlyk@users.noreply.github.com>
Date: Mon, 4 Aug 2025 12:00:42 -0500
Subject: [PATCH] Export exception to nvbench namespace

---
 python/cuda/nvbench/__init__.py  | 3 +++
 python/cuda/nvbench/__init__.pyi | 3 ++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/python/cuda/nvbench/__init__.py b/python/cuda/nvbench/__init__.py
index 993ff05..4617e52 100644
--- a/python/cuda/nvbench/__init__.py
+++ b/python/cuda/nvbench/__init__.py
@@ -43,6 +43,9 @@ from cuda.nvbench._nvbench import (  # noqa: E402
 from cuda.nvbench._nvbench import (  # noqa: E402
     Launch as Launch,
 )
+from cuda.nvbench._nvbench import (  # noqa: E402
+    NVBenchRuntimeError as NVBenchRuntimeError,
+)
 from cuda.nvbench._nvbench import (  # noqa: E402
     State as State,
 )
diff --git a/python/cuda/nvbench/__init__.pyi b/python/cuda/nvbench/__init__.pyi
index 6fea984..a0bca3d 100644
--- a/python/cuda/nvbench/__init__.pyi
+++ b/python/cuda/nvbench/__init__.pyi
@@ -33,7 +33,7 @@ class CudaStream:
 
     Note
     ----
-        The class is not directly constructible.
+        The class is not user-constructible.
     """
     def __cuda_stream__(self) -> tuple[int, int]:
         """
@@ -65,6 +65,7 @@ class Benchmark:
     Note
     ----
         The class is not user-constructible.
+
         Use `~register` function to create Benchmark and register
         it with NVBench.
     """