From 7c6bc1dfe6c4d6b71e6eb79d54cc145798d109d5 Mon Sep 17 00:00:00 2001 From: CARFAC Team Date: Tue, 30 Jul 2024 22:45:20 -0700 Subject: [PATCH] Fix too-tight tolerance on CARFAC golden data test. PiperOrigin-RevId: 657867039 --- python/tf/carfac_golden_data_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tf/carfac_golden_data_test.py b/python/tf/carfac_golden_data_test.py index 121c8cb..f0ffc71 100644 --- 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__':