Fix too-tight tolerance on CARFAC golden data test.

PiperOrigin-RevId: 657867039
google · Jul 31, 2024 · 7c6bc1d · 7c6bc1d
1 parent f2f3984
commit 7c6bc1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tf/carfac_golden_data_test.py b/python/tf/carfac_golden_data_test.py
@@ -54,7 +54,7 @@ def testEagerModeAgainstGoldenData(self):
     np.savez(output_file, data=output[:, :, :, :, 0])
     print(f'Golden data test saved produced output in {output_file}')
    golden = np.load(pathlib.Path(__file__).parent / 'golden_data.npz')['data']
-    np.testing.assert_allclose(golden, output[:, :, :, :, 0])
+    np.testing.assert_allclose(golden, output[:, :, :, :, 0], atol=1e-6)
 
 
 if __name__ == '__main__':