From 8017c7a5dff0c8c062f7e25f5ce9a9598642a48b Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Sat, 4 Jun 2022 12:13:25 +0800 Subject: [PATCH] Release GIL for online feature extraction. --- kaldifeat/python/csrc/online-feature.cc | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/kaldifeat/python/csrc/online-feature.cc b/kaldifeat/python/csrc/online-feature.cc index 13e4a4f..2d1dcd8 100644 --- a/kaldifeat/python/csrc/online-feature.cc +++ b/kaldifeat/python/csrc/online-feature.cc @@ -22,9 +22,11 @@ void PybindOnlineFeatureTpl(py::module &m, const std::string &class_name, .def_property_readonly("num_frames_ready", &PyClass::NumFramesReady) .def("is_last_frame", &PyClass::IsLastFrame, py::arg("frame")) .def("get_frame", &PyClass::GetFrame, py::arg("frame")) - .def("get_frames", &PyClass::GetFrames, py::arg("frames")) + .def("get_frames", &PyClass::GetFrames, py::arg("frames"), + py::call_guard()) .def("accept_waveform", &PyClass::AcceptWaveform, - py::arg("sampling_rate"), py::arg("waveform")) + py::arg("sampling_rate"), py::arg("waveform"), + py::call_guard()) .def("input_finished", &PyClass::InputFinished); }