diff --git a/tests/test_app.py b/tests/test_app.py index 8e10ee99..0a309653 100644 --- a/tests/test_app.py +++ b/tests/test_app.py @@ -131,7 +131,7 @@ def test_varBayes(filename, expected, request): arr_1 = np.array(varBayes.iter_delta, dtype=np.float32).round(11) arr_2 = np.array(expected_iter_delta[: cfg['max_iter']], dtype=np.float32).round(11) # print(f"max diff: {np.abs(arr_1-arr_2).max()}") - # assert np.allclose(arr_1, arr_2, rtol=0, atol=1e-06) - assert np.all(arr_1 == arr_2) + assert np.allclose(arr_1, arr_2, rtol=1e-05, atol=1e-06) + # assert np.all(arr_1 == arr_2)