Skip to content

Commit bbf563c

Browse files
committed
fix test
1 parent 41f4ebc commit bbf563c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

_unittests/ut_tasks/try_export.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -222,7 +222,7 @@ def _config_reduction(config, task):
222222
self.assertIn('"PackedMultiHeadAttention"', str(model))
223223
elif attention == "BIGMASK":
224224
self.assertNotIn('"PackedMultiHeadAttention"', str(model))
225-
self.assertNotIn("MultiHeadAttention", str(model))
225+
self.assertIn("MultiHeadAttention", str(model))
226226
self.assertNotIn("Loop", {n.op_type for n in model.graph.node})
227227
elif attention == "LOOPMHA":
228228
self.assertNotIn('"PackedMultiHeadAttention"', str(model))

0 commit comments

Comments
 (0)