diff --git a/functionary/train/train.py b/functionary/train/train.py
index a3bf8db7..0fd319fc 100644
--- a/functionary/train/train.py
+++ b/functionary/train/train.py
@@ -140,6 +140,8 @@ def trainer_save_model_safe(trainer: transformers.Trainer):
 
 
 def train():
+    """Training loop"""
+
     argument_parser = transformers.HfArgumentParser(
         (ModelArguments, DataArguments, TrainingArguments)
     )