diff --git a/scripts/finetune.py b/scripts/finetune.py
index cd9234334..2f6bef3ef 100644
--- a/scripts/finetune.py
+++ b/scripts/finetune.py
@@ -261,7 +261,6 @@ def train(
         model.save_pretrained(cfg.output_dir)
         return
 
-
     trainer = setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer)
 
     model.config.use_cache = False
diff --git a/src/axolotl/utils/trainer.py b/src/axolotl/utils/trainer.py
index 57a08aa53..b7823fea4 100644
--- a/src/axolotl/utils/trainer.py
+++ b/src/axolotl/utils/trainer.py
@@ -1,6 +1,7 @@
 """Module containing the Trainer class and related functions"""
 
 import importlib
+import logging
 import math
 import os
 import sys