fix ut

xadupre · xadupre · commit ee2579a66fd8 · 2024-03-26T12:03:56.000+01:00
diff --git a/_unittests/ut_light_api/test_backend_export.py b/_unittests/ut_light_api/test_backend_export.py
@@ -242,7 +242,7 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):
 
 # The following tests are too slow with the reference implementation (Conv).
 backend_test.exclude(
-    "(FLOAT8|BFLOAT16|_opt_|_3d_|_momentum_|_4d_"
+    "(FLOAT8|BFLOAT16|INT4|_opt_|_3d_|_momentum_|_4d_"
     "|test_adagrad"
     "|test_adam"
     "|test_ai_onnx_ml_"
diff --git a/_unittests/ut_reference/test_backend_extended_reference_evaluator.py b/_unittests/ut_reference/test_backend_extended_reference_evaluator.py
@@ -149,7 +149,7 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):
     "|test_scan_sum)"
 )
 
-if onnx_opset_version() < 21:
+if onnx_opset_version() < 200:
     # The following tests are using types not supported by NumPy.
     # They could be if method to_array is extended to support custom
     # types the same as the reference implementation does

Original file line number	Diff line number	Diff line change
`@@ -149,7 +149,7 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):`
`149`	`149`	`"\|test_scan_sum)"`
`150`	`150`	`)`
`151`	`151`
`152`		`-if onnx_opset_version() < 21:`
	`152`	`+if onnx_opset_version() < 200:`
`153`	`153`	`# The following tests are using types not supported by NumPy.`
`154`	`154`	`# They could be if method to_array is extended to support custom`
`155`	`155`	`# types the same as the reference implementation does`