From 505009b45466a0444ab3f85667ee2e2d004a2105 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sun, 25 May 2025 16:53:20 -0400 Subject: [PATCH] add close to comment block --- src/axolotl/core/builders/causal.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/axolotl/core/builders/causal.py b/src/axolotl/core/builders/causal.py index 8babf6a65..e2ad1f579 100644 --- a/src/axolotl/core/builders/causal.py +++ b/src/axolotl/core/builders/causal.py @@ -130,6 +130,9 @@ class HFCausalTrainerBuilder(TrainerBuilderBase): return callbacks def _get_trainer_cls(self): + """ + Gets the trainer class for the given configuration. + """ if self.cfg.plugins: plugin_manager = PluginManager.get_instance() trainer_cls = plugin_manager.get_trainer_cls(self.cfg)