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)