diff options
| -rw-r--r-- | mypy.ini | 5 | ||||
| -rw-r--r-- | swr2_asr/train.py | 2 |
2 files changed, 5 insertions, 2 deletions
@@ -2,4 +2,7 @@ ignore_missing_imports = True [mypy-torchaudio.*] -ignore_missing_imports = true
\ No newline at end of file +ignore_missing_imports = true + +[mypy-torch.*] +ignore_missing_imports = true diff --git a/swr2_asr/train.py b/swr2_asr/train.py index df9ffa2..7a9ffec 100644 --- a/swr2_asr/train.py +++ b/swr2_asr/train.py @@ -1,6 +1,6 @@ """Training script for the ASR model.""" -from AudioLoader.speech.mls import MultilingualLibriSpeech import os +from AudioLoader.speech.mls import MultilingualLibriSpeech def main() -> None: |