Skip to content

Commit 2e8f3ac

Browse files
author
varun sundar rabindranath
committed
fix lint
Signed-off-by: varun sundar rabindranath <[email protected]>
1 parent fda5f44 commit 2e8f3ac

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/kernels/test_cutlass_moe.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -234,7 +234,10 @@ def run_16_bit(moe_tensors: MOETensors,
234234
if not with_ep:
235235
return cutlass_moe(**kwargs)
236236

237-
return run_with_expert_maps(num_experts, num_local_experts, **kwargs)
237+
return run_with_expert_maps(
238+
num_experts,
239+
num_local_experts, # type: ignore[arg-type]
240+
**kwargs)
238241

239242

240243
@pytest.mark.parametrize("m,n,k", MNK_FACTORS)

0 commit comments

Comments
 (0)