Support init FasterDecoder with StdVectorFst and StdConstFst. (#4)

k2-fsa · Oct 2, 2023 · 5d3ef5b · 5d3ef5b
1 parent a14454f
commit 5d3ef5b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -9,7 +9,7 @@ if (CMAKE_VERSION VERSION_GREATER_EQUAL "3.24.0")
   cmake_policy(SET CMP0135 NEW)
 endif()
 
-set(KALDI_DECODER_VERSION "0.2.1")
+set(KALDI_DECODER_VERSION "0.2.2")
 
 set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY "${CMAKE_BINARY_DIR}/lib")
 set(CMAKE_LIBRARY_OUTPUT_DIRECTORY "${CMAKE_BINARY_DIR}/lib")

diff --git a/kaldi-decoder/python/csrc/faster-decoder.cc b/kaldi-decoder/python/csrc/faster-decoder.cc
@@ -34,6 +34,12 @@ void PybindFasterDecoder(py::module *m) {
       .def(py::init<const fst::Fst<fst::StdArc> &,
                     const FasterDecoderOptions &>(),
            py::arg("fst"), py::arg("config"))
+      .def(py::init<const fst::VectorFst<fst::StdArc> &,
+                    const FasterDecoderOptions &>(),
+           py::arg("fst"), py::arg("config"))
+      .def(py::init<const fst::ConstFst<fst::StdArc> &,
+                    const FasterDecoderOptions &>(),
+           py::arg("fst"), py::arg("config"))
       .def("set_options", &PyClass::SetOptions, py::arg("config"))
       .def("decode", &PyClass::Decode, py::arg("decodable"))
       .def("reached_final", &PyClass::ReachedFinal)