From b6821b7624dde5bf04097d0334c9560ee73e3337 Mon Sep 17 00:00:00 2001
From: Oleksandr Pavlyk <21087696+oleksandr-pavlyk@users.noreply.github.com>
Date: Fri, 25 Jul 2025 16:58:02 -0500
Subject: [PATCH] Rename NVBenchRuntimeException to NVBenchRuntimeError

Added exception to __init__.pyi
---
 python/cuda/nvbench/__init__.pyi | 5 +++++
 python/src/py_nvbench.cpp        | 2 +-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/python/cuda/nvbench/__init__.pyi b/python/cuda/nvbench/__init__.pyi
index 607918f..5e7b390 100644
--- a/python/cuda/nvbench/__init__.pyi
+++ b/python/cuda/nvbench/__init__.pyi
@@ -261,3 +261,8 @@ def run_all_benchmarks(argv: Sequence[str]) -> None:
         Sequence of CLI arguments controlling NVBench. Usually, it is `sys.argv`.
     """
     ...
+
+class NVBenchRuntimeError(RuntimeError):
+    """An exception raised if running benchmarks encounters an error"""
+
+    ...
diff --git a/python/src/py_nvbench.cpp b/python/src/py_nvbench.cpp
index 8577fe9..a3cba65 100644
--- a/python/src/py_nvbench.cpp
+++ b/python/src/py_nvbench.cpp
@@ -595,7 +595,7 @@ PYBIND11_MODULE(_nvbench, m)
   // Use handle to take a memory leak here, since this object's destructor may be called after
   // interpreter has shut down
   benchmark_exc =
-    py::exception<nvbench_run_error>(m, "NVBenchRuntimeException", PyExc_RuntimeError).release();
+    py::exception<nvbench_run_error>(m, "NVBenchRuntimeError", PyExc_RuntimeError).release();
   // == STEP 6
   //    ATTN: nvbench::benchmark_manager is a singleton