Commit 8a885191 authored by Caroline Chen's avatar Caroline Chen Committed by Facebook GitHub Bot
Browse files

Fix type for lm parameter in decoder (#2273)

Summary: Pull Request resolved: https://github.com/pytorch/audio/pull/2273

Reviewed By: mthrok

Differential Revision: D34799335

Pulled By: carolineechen

fbshipit-source-id: d0eea79448efdbd84758a3f433ab9350b4c94e91
parent ee68fcc5
...@@ -181,7 +181,7 @@ class LexiconDecoder: ...@@ -181,7 +181,7 @@ class LexiconDecoder:
def lexicon_decoder( def lexicon_decoder(
lexicon: str, lexicon: str,
tokens: Union[str, List[str]], tokens: Union[str, List[str]],
lm: str = None, lm: Optional[str] = None,
nbest: int = 1, nbest: int = 1,
beam_size: int = 50, beam_size: int = 50,
beam_size_token: Optional[int] = None, beam_size_token: Optional[int] = None,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment