From 763a51486ed1aa58c3808e9438832e6cb3c510d2 Mon Sep 17 00:00:00 2001 From: Askars <19928242+vsd-vector@users.noreply.github.com> Date: Tue, 20 Feb 2024 14:47:53 +0200 Subject: [PATCH] Add missing start_time to python API (#591) Co-authored-by: vsd-vector --- sherpa-onnx/python/csrc/online-recognizer.cc | 3 +++ sherpa-onnx/python/sherpa_onnx/online_recognizer.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/sherpa-onnx/python/csrc/online-recognizer.cc b/sherpa-onnx/python/csrc/online-recognizer.cc index c40b541c..c10cb3b4 100644 --- a/sherpa-onnx/python/csrc/online-recognizer.cc +++ b/sherpa-onnx/python/csrc/online-recognizer.cc @@ -23,6 +23,9 @@ static void PybindOnlineRecognizerResult(py::module *m) { .def_property_readonly( "tokens", [](PyClass &self) -> std::vector { return self.tokens; }) + .def_property_readonly( + "start_time", + [](PyClass &self) -> float { return self.start_time; }) .def_property_readonly( "timestamps", [](PyClass &self) -> std::vector { return self.timestamps; }); diff --git a/sherpa-onnx/python/sherpa_onnx/online_recognizer.py b/sherpa-onnx/python/sherpa_onnx/online_recognizer.py index c92b605f..70de7a2a 100644 --- a/sherpa-onnx/python/sherpa_onnx/online_recognizer.py +++ b/sherpa-onnx/python/sherpa_onnx/online_recognizer.py @@ -509,6 +509,9 @@ class OnlineRecognizer(object): def timestamps(self, s: OnlineStream) -> List[float]: return self.recognizer.get_result(s).timestamps + def start_time(self, s: OnlineStream) -> float: + return self.recognizer.get_result(s).start_time + def is_endpoint(self, s: OnlineStream) -> bool: return self.recognizer.is_endpoint(s)