diff --git a/wonnx-py/tests/test_onnx_backend.py b/wonnx-py/tests/test_onnx_backend.py index 9d0c4a61..ed556149 100644 --- a/wonnx-py/tests/test_onnx_backend.py +++ b/wonnx-py/tests/test_onnx_backend.py @@ -144,6 +144,7 @@ def do_enforce_test_coverage_safelist(model): # type: (ModelProto) -> bool backend_test.include(f"test_reciprocal_[a-z,_]*") backend_test.include(f"test_shape_[a-z,_]*") backend_test.include(f"test_size_[a-z,_]*") +backend_test.include(f"test_celu_[a-z,_]*") # For these we only test the default version, as we don't support the bool type backend_test.include(f"test_prelu_broadcast_cpu$") @@ -151,10 +152,6 @@ def do_enforce_test_coverage_safelist(model): # type: (ModelProto) -> bool backend_test.include(f"test_relu_cpu$") backend_test.include(f"test_leakyrelu_default_cpu$") - -# Fails due to tolerance issues? -# backend_test.include(f"test_celu_cpu$") - # Don't support 'bool' type # backend_test.include(f"test_and_bcast[a-z0-9,_]*") # backend_test.include(f"test_or_bcast[a-z0-9,_]*")