Skip to content

Commit 680ed17

Browse files
committed
[tmva][test] Increase tolerance for gnn test from 5 to 4 decimals
1 parent 2c7ffbd commit 680ed17

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

bindings/pyroot/pythonizations/test/sofie_gnn.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -337,11 +337,11 @@ def test_c_lhcb_toy_inference(self):
337337
output_edge_data = output_gn[i].edges.numpy()
338338
output_global_data = output_gn[i].globals.numpy().flatten()
339339

340-
assert_almost_equal(output_node_data, np.asarray(output_ops[i].node_data), 5)
340+
assert_almost_equal(output_node_data, np.asarray(output_ops[i].node_data), 4)
341341

342-
assert_almost_equal(output_edge_data, np.asarray(output_ops[i].edge_data), 5)
342+
assert_almost_equal(output_edge_data, np.asarray(output_ops[i].edge_data), 4)
343343

344-
assert_almost_equal(output_global_data, np.asarray(output_ops[i].global_data), 5)
344+
assert_almost_equal(output_global_data, np.asarray(output_ops[i].global_data), 4)
345345

346346
@classmethod
347347
def tearDownClass(self):

0 commit comments

Comments
 (0)