From 247d958ef23a97bc6186c34c5671cee2ee373a41 Mon Sep 17 00:00:00 2001 From: emilyaf Date: Wed, 13 Dec 2023 17:27:27 -0800 Subject: [PATCH] Pin Bazel version to 6.4.0, while we address breakages caused by 7.0. PiperOrigin-RevId: 590762991 --- .../python/internal/backend/numpy/numpy_test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow_probability/python/internal/backend/numpy/numpy_test.py b/tensorflow_probability/python/internal/backend/numpy/numpy_test.py index 56c3641350..13140fe36c 100644 --- a/tensorflow_probability/python/internal/backend/numpy/numpy_test.py +++ b/tensorflow_probability/python/internal/backend/numpy/numpy_test.py @@ -1001,7 +1001,8 @@ def _not_implemented(*args, **kwargs): 'linalg.svd', [single_arrays( shape=shapes(min_dims=2), elements=floats(min_value=-1e10, max_value=1e10))], - post_processor=_svd_post_process), + post_processor=_svd_post_process, + atol=2e-5), TestCase( 'linalg.qr', [ qr_params(),