fix pre-commit

src/axolotl/core/trainer_builder.py

@@ -857,9 +857,7 @@ def get_callbacks(self) -> List[TrainerCallback]:
         if self.cfg.vessl_credential_path:
             from axolotl.utils.callbacks.vessl_ import VesslLogMetricsCallback
 
-            callbacks.append(
-                VesslLogMetricsCallback(self.cfg.vessl_credential_path)
-            )
+            callbacks.append(VesslLogMetricsCallback(self.cfg.vessl_credential_path))
 
         return callbacks
 

src/axolotl/utils/callbacks/vessl_.py

@@ -1,19 +1,27 @@
 """Vessl module for trainer callbacks"""
 import logging
-from typing import Dict, List
+from typing import Dict
 
 import vessl
 from transformers import TrainerCallback, TrainerControl, TrainerState
 from transformers.training_args import TrainingArguments
 
 LOG = logging.getLogger("axolotl.callbacks")
 
+
 class VesslLogMetricsCallback(TrainerCallback):
+    """Callback to send training metrics to VESSL AI"""
 
     def __init__(self, credential_path: str) -> None:
         vessl.configure(credentials_file=credential_path)
 
-    def on_log(self, args: TrainingArguments, state: TrainerState, control: TrainerControl, logs: Dict[str, float] = None, **kwargs):     
+    def on_log(
+        self,
+        args: TrainingArguments,  # pylint: disable=unused-argument
+        state: TrainerState,
+        control: TrainerControl,  # pylint: disable=unused-argument
+        logs: Dict[str, float],
+        **kwargs  # pylint: disable=unused-argument
+    ):
         if state.is_world_process_zero:
             vessl.log(logs, state.global_step)
-