From 2ac880f87004eaa47b53b046738d2708e46dc16b Mon Sep 17 00:00:00 2001 From: austereantelope <95935342+austereantelope@users.noreply.github.com> Date: Mon, 24 Jan 2022 13:07:21 -0600 Subject: [PATCH] tighten test_process_pixels_gray bound --- ml-agents-envs/mlagents_envs/tests/test_rpc_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ml-agents-envs/mlagents_envs/tests/test_rpc_utils.py b/ml-agents-envs/mlagents_envs/tests/test_rpc_utils.py index 1d03b21f411..f41421d66d9 100644 --- a/ml-agents-envs/mlagents_envs/tests/test_rpc_utils.py +++ b/ml-agents-envs/mlagents_envs/tests/test_rpc_utils.py @@ -258,7 +258,7 @@ def test_process_pixels_gray(): byte_arr = generate_compressed_data(in_array) out_array = process_pixels(byte_arr, 1) assert out_array.shape == (128, 64, 1) - assert np.mean(in_array.mean(axis=2, keepdims=True) - out_array) < 0.01 + assert np.mean(in_array.mean(axis=2, keepdims=True) - out_array) < 0.002 assert np.allclose(in_array.mean(axis=2, keepdims=True), out_array, atol=0.01)