fix ci pipeline

1 files changed, 3 insertions, 6 deletions

diff --git a/swr2_asr/train.py b/swr2_asr/train.py
index abe0e08..f61776a 100644
--- a/swr2_asr/train.py
+++ b/swr2_asr/train.py
@@ -1,12 +1,9 @@
-# import dataset
-# define model
-# define loss
-# define optimizer
-# train
+"""Training script for the ASR model."""
 from AudioLoader.speech.mls import MultilingualLibriSpeech
 
 
-def main():
+def main() -> None:
+    """Main function."""
     dataset = MultilingualLibriSpeech(
         "data", "mls_polish_opus", split="train", download=True
     )