Skip to content

Commit 4a481cf

Browse files
committed
fix
1 parent d89735e commit 4a481cf

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

_scripts/test_backend_onnxruntime.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -141,11 +141,8 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):
141141

142142
backend_test.exclude("(test_adagrad|test_adam|test_add_uint8)")
143143

144-
if pv.Version(onnxruntime.__version__) < pv.Version("1.24"):
145-
backend_test.exclude(
146-
"(test_attention_4d_with_qk|test_attention_4d_gqa|"
147-
"test_attention_4d_with_past_and_present_qk)"
148-
)
144+
if pv.Version(onnxruntime.__version__) <= pv.Version("1.24"):
145+
backend_test.exclude("(test_attention_4d_with|test_attention_4d_gqa)")
149146

150147

151148
# import all test cases at global scope to make them visible to python.unittest

0 commit comments

Comments
 (0)