From 2d6b537157af1ee2f65026b28a5258838e356f32 Mon Sep 17 00:00:00 2001 From: Matthew Bonanni Date: Fri, 23 Jan 2026 13:26:56 -0500 Subject: [PATCH] [Bugfix][CI] Fix pre-commit (#32956) Signed-off-by: Matthew Bonanni --- vllm/v1/worker/mamba_utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/vllm/v1/worker/mamba_utils.py b/vllm/v1/worker/mamba_utils.py index a0a1ae224..56fb02380 100644 --- a/vllm/v1/worker/mamba_utils.py +++ b/vllm/v1/worker/mamba_utils.py @@ -4,13 +4,12 @@ import itertools from typing import Any import torch -import triton -import triton.language as tl from vllm.config import CacheConfig from vllm.model_executor.layers.mamba.mamba_utils import ( MambaStateCopyFunc, ) +from vllm.triton_utils import tl, triton from vllm.v1.core.sched.output import SchedulerOutput from vllm.v1.kv_cache_interface import KVCacheConfig, MambaSpec from vllm.v1.worker.gpu_input_batch import CachedRequestState