Fixes too small tolerance in unit test.

PiperOrigin-RevId: 635489346
Change-Id: I54d3c8e4356bab2a4261771d583c3b689ad12bcd

Author: Johannes Ballé

tensorflow_compression/python/ops/quantization_ops_test.py

@@ -71,7 +71,7 @@ def test_rounding_is_unbiased(self):
     rounded = gen_ops.stochastic_round(replicated, 1., ())
     self.assertEqual(rounded.dtype, tf.int32)
     averaged = tf.reduce_mean(tf.cast(rounded, tf.float32), axis=0)
-    self.assertAllClose(values, averaged, atol=5e-3, rtol=0)
+    self.assertAllClose(values, averaged, atol=1e-2, rtol=0)
 
 
 if __name__ == "__main__":