fix python CI, missing type

k2-fsa · Sep 4, 2024 · 8faedba · 8faedba
1 parent 5aa9f82
commit 8faedba
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sherpa-onnx/python/csrc/online-lm-config.cc b/sherpa-onnx/python/csrc/online-lm-config.cc
@@ -16,8 +16,8 @@ void PybindOnlineLMConfig(py::module *m) {
       .def(py::init<const std::string &, float, int32_t,
            const std::string &, bool>(),
            py::arg("model") = "", py::arg("scale") = 0.5f,
-           py::arg("lm_num_threads") = 1, py::arg("lm_provider") = "cpu"),
-           py::arg("shallow_fusion") = true
+           py::arg("lm_num_threads") = 1, py::arg("lm_provider") = "cpu",
+           py::arg("shallow_fusion") = true)
       .def_readwrite("model", &PyClass::model)
       .def_readwrite("scale", &PyClass::scale)
       .def_readwrite("lm_provider", &PyClass::lm_provider)

diff --git a/sherpa-onnx/python/sherpa_onnx/online_recognizer.py b/sherpa-onnx/python/sherpa_onnx/online_recognizer.py
@@ -64,7 +64,7 @@ def from_transducer(
         bpe_vocab: str = "",
         lm: str = "",
         lm_scale: float = 0.1,
-        lm_shallow_fusion = True,
+        lm_shallow_fusion: bool = True,
         temperature_scale: float = 2.0,
         debug: bool = False,
         rule_fsts: str = "",