Skip to content

Commit f96c360

Browse files
fix mypy
1 parent 691952c commit f96c360

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onnxscript/rewriter/ort_fusions/attention.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,7 @@ def no_match(val: ir.Value, dims: Sequence[str]) -> bool:
131131
if Dh is None or Dh_q is None or Dh_k is None or Dh_v is None:
132132
return False # Missing bindings, cannot verify
133133

134-
if Dh != Dh_q + Dh_k + Dh_v:
134+
if Dh != Dh_q + Dh_k + Dh_v: # type: ignore[operator]
135135
return False # Hidden size mismatch
136136
# TODO: Add mask check once mask is added to the pattern
137137

0 commit comments

Comments
 (0)