Skip to content

Commit a3451c5

Browse files
committed
[Test] use np.isclose instead of equals for test condition
1 parent 55c2f6f commit a3451c5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/transformation/test_conv_lowering.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def test_conv_lowering_quant_weights(model_name):
5555
input_dict = {model.graph.input[0].name: input_t}
5656
prod_dict = oxe.execute_onnx(model, input_dict)
5757
prod_t = prod_dict[model.graph.output[0].name]
58-
assert (prod_t == golden_t).all()
58+
assert np.isclose(prod_t, golden_t).all()
5959

6060

6161
def test_conv_lowering_convmnist():

0 commit comments

Comments
 (0)