chore: fix formating of test_run_with_additional_callbacks

Signed-off-by: Vassilis Vassiliadis <vassilis.vassiliadis@ibm.com>

Author: VassilisVassiliadis

tests/test_sft_trainer.py

@@ -23,6 +23,7 @@
 
 # Third Party
 from datasets.exceptions import DatasetGenerationError
+from transformers.trainer_callback import TrainerCallback
 import pytest
 import torch
 import transformers
@@ -620,8 +621,6 @@ def test_bad_torch_dtype():
 
 def test_run_with_additional_callbacks():
     """Ensure that train() can work with additional_callbacks"""
-    # Third Party
-    from transformers.trainer_callback import TrainerCallback
 
     with tempfile.TemporaryDirectory() as tempdir:
         train_args = copy.deepcopy(TRAIN_ARGS)