diff --git a/core/general-client/src/pybind_general_model.cpp b/core/general-client/src/pybind_general_model.cpp index 47bc6bd3308e5150ccaba29ccefc52ca6e177c64..c8a0e3e612dc68603829a99e9116a4b8ae6207e3 100644 --- a/core/general-client/src/pybind_general_model.cpp +++ b/core/general-client/src/pybind_general_model.cpp @@ -83,7 +83,8 @@ PYBIND11_MODULE(serving_client, m) { fetch_name, predict_res, pid); - }) + }, + py::call_guard()) .def("batch_predict", [](PredictorClient &self, const std::vector>> @@ -102,7 +103,8 @@ PYBIND11_MODULE(serving_client, m) { fetch_name, predict_res_batch, pid); - }); + }, + py::call_guard()); } } // namespace general_model