Adjusts tolerance of a unit test slightly.

PiperOrigin-RevId: 298366310
Change-Id: I9648e2e6b532b4b65686797812ba953efb34d04e

Author: Johannes Ballé

tensorflow_compression/python/layers/entropy_models_test.py

@@ -237,7 +237,7 @@ def test_normalization(self):
       likelihood, = sess.run([likelihood], {inputs: x})
     self.assertEqual(x.shape, likelihood.shape)
     integral = np.sum(likelihood) * .0005
-    self.assertAllClose(1, integral, rtol=0, atol=1e-4)
+    self.assertAllClose(1, integral, rtol=0, atol=2e-4)
 
   def test_entropy_estimates(self):
     # Test that entropy estimates match actual range coding.