add st unit test

chintu619 · Apr 27, 2022 · 72b6b21 · 72b6b21
1 parent d1e8ac3
commit 72b6b21
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/espnet2/st/espnet_model.py b/espnet2/st/espnet_model.py
@@ -53,9 +53,9 @@ def __init__(
         decoder: AbsDecoder,
         extra_asr_decoder: Optional[AbsDecoder],
         extra_mt_decoder: Optional[AbsDecoder],
-        ctc: CTC,
-        src_vocab_size: int = 0,
-        src_token_list: Union[Tuple[str, ...], List[str]] = [],
+        ctc: Optional[CTC],
+        src_vocab_size: Optional[int],
+        src_token_list: Optional[Union[Tuple[str, ...], List[str]]],
         asr_weight: float = 0.0,
         mt_weight: float = 0.0,
         mtlalpha: float = 0.0,