add close to comment block

2025-05-25 16:53:20 -04:00
parent b4e96ef12c
commit 505009b454
1 changed files with 3 additions and 0 deletions
--- 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)