Relaxing census v2 prediction test to atol=0.01

PiperOrigin-RevId: 511870674

Author: zoyahav

examples/census_example_v2_test.py

@@ -176,7 +176,7 @@ def testCensusExampleAccuracy(self, read_raw_data_for_training):
     prediction = model.signatures['serving_default'](
         tf.constant([example.SerializeToString()], tf.string))
     self.assertAllEqual([['0', '1']], prediction['classes'])
-    self.assertAllClose([[0, 1]], prediction['scores'], atol=0.001)
+    self.assertAllClose([[0, 1]], prediction['scores'], atol=0.01)
 
     # This is required in order to support the classify API for this Keras
     # model.