Skip to content

Commit b528ccc

Browse files
committed
gitnore type for precommit
Signed-off-by: Ye Yu <[email protected]>
1 parent b47928e commit b528ccc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modelopt/torch/speculative/plugins/megatron_eagle.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -835,7 +835,7 @@ def _get_eagle_module_inputs(
835835
dtype=hidden_states.dtype,
836836
device=hidden_states.device,
837837
),
838-
gathered_features[:-1, :, :],
838+
gathered_features[:-1, :, :], # type: ignore[index]
839839
)
840840
)
841841
)

0 commit comments

Comments
 (0)