From 3f36cb5e88090495fd46c2e4f04cab61867cd234 Mon Sep 17 00:00:00 2001
From: Musab Gultekin <87330355+musab-mk@users.noreply.github.com>
Date: Fri, 22 Nov 2024 15:21:50 -0700
Subject: [PATCH] Update train.py

---
 functionary/train/train.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/functionary/train/train.py b/functionary/train/train.py
index a3bf8db..0fd319f 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)
     )