From bf834974755c1241e0b6cd9609c91b50d4819456 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Thu, 9 Nov 2023 12:44:02 +0800 Subject: [PATCH] typo fixes --- kaldifeat/python/csrc/utils.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/kaldifeat/python/csrc/utils.cc b/kaldifeat/python/csrc/utils.cc index 2acab71..f7a6fc2 100644 --- a/kaldifeat/python/csrc/utils.cc +++ b/kaldifeat/python/csrc/utils.cc @@ -130,7 +130,7 @@ WhisperFbankOptions WhisperFbankOptionsFromDict(py::dict dict) { opts.frame_opts = FrameExtractionOptionsFromDict(dict["frame_opts"]); } - AS_DICT(num_mels); + FROM_DICT(int_, num_mels); if (dict.contains("device")) { opts.device = torch::Device(std::string(py::str(dict["device"]))); @@ -144,6 +144,8 @@ py::dict AsDict(const WhisperFbankOptions &opts) { dict["frame_opts"] = AsDict(opts.frame_opts); + AS_DICT(num_mels); + auto torch_device = py::module_::import("torch").attr("device"); dict["device"] = torch_device(opts.device.str());