From 2d259159808fc5370933a3843abba8b1f4560c1d Mon Sep 17 00:00:00 2001
From: Pherkel
Date: Fri, 18 Aug 2023 13:57:03 +0200
Subject: fix ci pipeline

---
 swr2_asr/train.py | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

(limited to 'swr2_asr/train.py')

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
     )
-- 
cgit v1.2.3