From 8c1f8da225ba0c17747f999381624eb5e6316c60 Mon Sep 17 00:00:00 2001 From: Tommy van der Vorst Date: Sun, 26 Mar 2023 18:03:00 +0200 Subject: [PATCH] fix(tests): enable ONNX backend test case for Celu operator --- wonnx-py/tests/test_onnx_backend.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) 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,_]*")