ServicePyAdapter::pivot - release GIL, redirect stdout/stderr

browsermt · Jan 25, 2022 · c51ae04 · c51ae04
1 parent da699f3
commit c51ae04
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/bindings/python/bergamot.cpp b/bindings/python/bergamot.cpp
@@ -89,6 +89,16 @@ class ServicePyAdapter {
   std::vector<Response> pivot(Model first, Model second,
                               std::vector<std::string> &inputs,
                               const ResponseOptions &options) {
+    py::scoped_ostream_redirect outstream(
+        std::cout,                                // std::ostream&
+        py::module_::import("sys").attr("stdout") // Python output
+    );
+    py::scoped_ostream_redirect errstream(
+        std::cerr,                                // std::ostream&
+        py::module_::import("sys").attr("stderr") // Python output
+    );
+
+    py::call_guard<py::gil_scoped_release> gil_guard;
     // Prepare promises, save respective futures. Have callback's in async set
     // value to the promises.
     std::vector<std::future<Response>> futures;