diff --git a/src/sparseml/transformers/integration_helper_functions.py b/src/sparseml/transformers/integration_helper_functions.py
index d7ca205f581..d26256afbad 100644
--- a/src/sparseml/transformers/integration_helper_functions.py
+++ b/src/sparseml/transformers/integration_helper_functions.py
@@ -47,6 +47,7 @@
 from sparseml.transformers.utils.load_task_dataset import load_task_dataset
 from sparseml.transformers.utils.optimizations import apply_kv_cache_injection
 
+
 _LOGGER = logging.getLogger(__name__)