Skip to content

Commit

Permalink
fix(tests): enable ONNX backend test case for Celu operator
Browse files Browse the repository at this point in the history
  • Loading branch information
pixelspark committed Mar 26, 2023
1 parent 90fc8ba commit 8c1f8da
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions wonnx-py/tests/test_onnx_backend.py
Original file line number Diff line number Diff line change
Expand Up @@ -144,17 +144,14 @@ 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$")
backend_test.include(f"test_elu_cpu$")
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,_]*")
Expand Down

0 comments on commit 8c1f8da

Please sign in to comment.