added return 0 at the end of __call__ function for hydra.

Author: kohankhaki

templates/src/llm/finetune_distributed/train.py

@@ -210,7 +210,7 @@ def _build_training_arguments(
 
         return TrainingArguments(**kwargs)
 
-    def __call__(self, cfg: DictConfig):
+    def __call__(self, cfg: DictConfig) -> int:
         """Execute fine-tuning, handling checkpoints, training, and evaluation."""
         out_dir = Path(cfg.paths.out_dir)
         out_dir.mkdir(parents=True, exist_ok=True)
@@ -277,6 +277,8 @@ def __call__(self, cfg: DictConfig):
             "All done. Train metrics: %s Eval metrics: %s", metrics, eval_metrics
         )
 
+        return 0
+
     def checkpoint(
         self, *args: Any, **kwargs: Any
     ) -> submitit.helpers.DelayedSubmission: