mypy

vllm-project · Jan 6, 2025 · 7ad3a70 · 7ad3a70
1 parent 2161b73
commit 7ad3a70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/model_executor/layers/fused_moe/moe_torch_iterative.py b/vllm/model_executor/layers/fused_moe/moe_torch_iterative.py
@@ -48,4 +48,4 @@ def fused_moe(
         else:
             final_hidden_states = final_hidden_states + current_hidden_states
 
-    return final_hidden_states.view(orig_shape)
+    return final_hidden_states.view(orig_shape)  # type: ignore