Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

adding tolerance to flaky test #13850

Merged
merged 3 commits into from
Jan 14, 2019
Merged
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion tests/python-pytest/onnx/test_node.py
Original file line number Diff line number Diff line change
Expand Up @@ -204,7 +204,8 @@ def test_imports(self):
onnx_model = get_onnx_graph(test_name, names, input_tensors, onnx_name, output_shape, attrs)
bkd_rep = backend.prepare(onnx_model, operation='import')
mxnet_out = bkd_rep.run(inputs)
npt.assert_almost_equal(np_out, mxnet_out)
npt.assert_almost_equal(np_out, mxnet_out, decimal=4)


# test_case = ("test_case_name", mxnet op, "ONNX_op_name", [input_list], attribute map, MXNet_specific=True/False,
# fix_attributes = {'modify': {mxnet_attr_name: onnx_attr_name},
Expand Down