From 958da703762b7759eabdaa6fd7fad231228e1ad9 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Sat, 10 Jun 2023 15:28:08 -0400
Subject: [PATCH] fix formatting

---
 scripts/finetune.py          | 1 -
 src/axolotl/utils/trainer.py | 1 +
 2 files changed, 1 insertion(+), 1 deletion(-)

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