mypy

Signed-off-by: Avshalom <[email protected]>
vllm-project · Jan 7, 2025 · e81505a · e81505a
1 parent 7970de8
commit e81505a
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