mypy

Signed-off-by: NickLucche <nlucches@redhat.com>

Author: NickLucche

vllm/model_executor/layers/mamba/abstract.py

@@ -44,7 +44,7 @@ def get_attn_backend(self) -> type["AttentionBackend"]:
         pass
 
     @abstractmethod
-    def get_state_dtype(self) -> tuple[torch.dtype]:
+    def get_state_dtype(self) -> tuple[torch.dtype, ...]:
         pass
 
     def get_kv_cache_spec(self, vllm_config: VllmConfig) -> Optional[KVCacheSpec]: