diff --git a/deepspeed/compression/compress.py b/deepspeed/compression/compress.py
index 00bf488e9b078..a4194a76e0bc7 100644
--- a/deepspeed/compression/compress.py
+++ b/deepspeed/compression/compress.py
@@ -16,6 +16,7 @@
 except ImportError as e:
     nc = None
 
+
 def check_deepspeed_config(config):
     if isinstance(config, dict):
         return config
@@ -126,7 +127,6 @@ def init_compression(model, deepspeed_config, teacher_model=None, mpu=None):
     if shared_parameters[SPARSE_PRUNING_ENABLED] and \
         shared_parameters[SPARSE_PRUNING_METHOD] == SPARSE_PRUNING_METHOD_SNIP_MOMENTUM:
 
-
         assert nc is not None, "please ensure the neural_compressor python package is installed by pip or conda if user wants to use snip_momentum sparse pruning"
 
         from .helper import generate_pruners, register_on_step_begin