From 13ad115ca3063f3f890dd56bb046954cc1ad6bba Mon Sep 17 00:00:00 2001
From: Oleksandr Pavlyk <21087696+oleksandr-pavlyk@users.noreply.github.com>
Date: Tue, 22 Jul 2025 12:07:07 -0500
Subject: [PATCH] Add nvbench.Benchmark.set_run_once method

---
 python/src/py_nvbench.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/python/src/py_nvbench.cpp b/python/src/py_nvbench.cpp
index 822f9e1..7cdb24f 100644
--- a/python/src/py_nvbench.cpp
+++ b/python/src/py_nvbench.cpp
@@ -299,6 +299,13 @@ PYBIND11_MODULE(_nvbench, m)
       return std::ref(self);
     },
     py::return_value_policy::reference);
+  py_benchmark_cls.def(
+    "set_run_once",
+    [](nvbench::benchmark_base &self, bool v) {
+      self.set_run_once(v);
+      return std::ref(self);
+    },
+    py::return_value_policy::reference);
 
   // == STEP 5
   // Define PyState class