pre-commit fix

Signed-off-by: Akash kaothalkar <akash.kaothalkar@ibm.com>

Author: Akash kaothalkar

vllm/model_executor/layers/mamba/ops/cpu/gdn_attention.py

@@ -11,6 +11,8 @@
 from vllm.model_executor.layers.mamba.mamba_utils import is_conv_state_dim_first
 from vllm.model_executor.layers.mamba.ops.cpu.causal_conv1d import (
     causal_conv1d_fn_cpu as causal_conv1d_torch,
+)
+from vllm.model_executor.layers.mamba.ops.cpu.causal_conv1d import (
     causal_conv1d_update_cpu,
 )
 from vllm.utils.torch_utils import (
@@ -658,4 +660,3 @@ def register_cpu_gdn_attention_ops() -> None:
         fake_impl=cpu_gdn_attention_core_fake,
     )
     _CPU_GDN_ATTENTION_OPS_REGISTERED = True
-

vllm/model_executor/layers/mamba/short_conv.py

@@ -95,6 +95,8 @@ def forward_native(
         # for causal conv can be plugged in here later.
         from vllm.model_executor.layers.mamba.ops.cpu.causal_conv1d import (
             causal_conv1d_fn_cpu as causal_conv1d_torch,
+        )
+        from vllm.model_executor.layers.mamba.ops.cpu.causal_conv1d import (
             causal_conv1d_update_cpu,
         )
 
@@ -164,7 +166,7 @@ def forward_native(
         if has_decode:
             assert attn_metadata.state_indices_tensor_d is not None
             state_indices_d = attn_metadata.state_indices_tensor_d.flatten()
-            Bx_d = (B_d * x_d)  # (num_decodes, dim)
+            Bx_d = B_d * x_d  # (num_decodes, dim)
             out_d = causal_conv1d_update_cpu(
                 Bx_d,
                 conv_state,
@@ -341,4 +343,3 @@ def short_conv_fake(
     mutates_args=["output"],
     fake_impl=short_conv_fake,
 )
-