Bump tolerances of two tests to fix the TFP build.

PiperOrigin-RevId: 407494411
sakira · Nov 4, 2021 · 57ccb0f · 57ccb0f
1 parent b6e84e4
commit 57ccb0f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorflow_probability/python/layers/weight_norm_test.py b/tensorflow_probability/python/layers/weight_norm_test.py
@@ -310,7 +310,8 @@ def testConv2DInitializedCorrectly(self, transpose):
         self.conv_random_input, self.num_conv_filters, transpose=transpose)
 
     self.assertAllClose(true_init_g, self.evaluate(norm_layer.g))
-    self.assertAllClose(true_init_bias, self.evaluate(norm_layer.layer.bias))
+    self.assertAllClose(
+        true_init_bias, self.evaluate(norm_layer.layer.bias), rtol=5e-6)
 
   def testCheckpoint(self):
     model = self._define_model('sequential', self.data_dim, self.num_hidden)