Fix Python ScopedThreadEnvironment

mitsuba-renderer · Aug 12, 2024 · 8bac8e9 · 8bac8e9
1 parent 3fc8d54
commit 8bac8e9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/core/python/thread.cpp b/src/core/python/thread.cpp
@@ -112,6 +112,7 @@ MI_PY_EXPORT(Thread) {
                                             D(ScopedSetThreadEnvironment))
         .def(nb::init<const ThreadEnvironment &>())
         .def("__enter__", &PyScopedSetThreadEnvironment::enter)
-        .def("__exit__", &PyScopedSetThreadEnvironment::exit);
+        .def("__exit__", &PyScopedSetThreadEnvironment::exit,
+             "exc_type"_a.none(), "exc_val"_a.none(), "exc_tb"_a.none());
 }
 
diff --git a/src/core/tests/test_thread.py b/src/core/tests/test_thread.py
@@ -0,0 +1,7 @@
+import pytest
+import mitsuba as mi
+
+def test01_use_scoped_thread_environment(variant_scalar_rgb):
+    environment = mi.ThreadEnvironment()
+    with mi.ScopedSetThreadEnvironment(environment):
+        mi.Log(mi.LogLevel.Info, "Log from a thread environment.")