You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by GitBox <gi...@apache.org> on 2019/01/14 07:06:33 UTC
[incubator-mxnet] Diff for: [GitHub] nswamy merged pull request #13850:
adding tolerance to flaky test
diff --git a/tests/python-pytest/onnx/test_node.py b/tests/python-pytest/onnx/test_node.py
index 186666eb247..25fe9c9f9a5 100644
--- a/tests/python-pytest/onnx/test_node.py
+++ b/tests/python-pytest/onnx/test_node.py
@@ -204,7 +204,7 @@ 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},
With regards,
Apache Git Services