diff --git a/src/axolotl/core/training_args_base.py b/src/axolotl/core/training_args_base.py
index fbd387e5f..a158bc4a8 100644
--- a/src/axolotl/core/training_args_base.py
+++ b/src/axolotl/core/training_args_base.py
@@ -7,8 +7,6 @@ from typing import Optional
 
 from PIL.Image import Resampling
 
-from axolotl.utils.schemas.enums import RingAttnFunc
-
 
 @dataclass
 class AxolotlTrainingMixins:
diff --git a/src/axolotl/integrations/base.py b/src/axolotl/integrations/base.py
index 001436152..91b380138 100644
--- a/src/axolotl/integrations/base.py
+++ b/src/axolotl/integrations/base.py
@@ -22,7 +22,6 @@ from __future__ import annotations
 
 import collections
 import importlib
-import logging
 import traceback
 from typing import TYPE_CHECKING, Callable, OrderedDict, Union
 
@@ -165,7 +164,6 @@ class BasePlugin:
             trainer: The trainer object for training.
         """
 
-
     def get_training_args(self, cfg: DictDefault):  # pylint: disable=unused-argument):
         """
         Returns custom training arguments to set on TrainingArgs.
@@ -178,7 +176,7 @@ class BasePlugin:
         """
 
     def get_collator_cls_and_kwargs(
-        self, cfg: DictDefault, is_eval: bool=False
+        self, cfg: DictDefault, is_eval: bool = False
     ):  # pylint: disable=unused-argument):
         """
         Returns a custom class for the collator.