From 57e62ca03ce66370baec3e02f1aa31859648a52f Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Tue, 2 May 2023 14:10:30 -0700 Subject: [PATCH] Reenable scipy_stats_test in CI. Disable testTruncNormPdf on CPU, which is failing after an LLVM update. PiperOrigin-RevId: 528884880 --- tests/BUILD | 1 - tests/scipy_stats_test.py | 4 +++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/BUILD b/tests/BUILD index cc0aa0a65..27ecbe957 100644 --- a/tests/BUILD +++ b/tests/BUILD @@ -727,7 +727,6 @@ jax_test( }, tags = [ "noasan", - "notap", # TODO(b/271883906): Failing on xla.opt.debug.tpu_hw tap "notsan", ], # Times out ) diff --git a/tests/scipy_stats_test.py b/tests/scipy_stats_test.py index f3cf6fa30..d5efd578d 100644 --- a/tests/scipy_stats_test.py +++ b/tests/scipy_stats_test.py @@ -12,9 +12,9 @@ # See the License for the specific language governing permissions and # limitations under the License. - from functools import partial import itertools +import unittest from absl.testing import absltest @@ -767,6 +767,8 @@ class LaxBackedScipyStatsTests(jtu.JaxTestCase): @genNamedParametersNArgs(5) def testTruncnormPdf(self, shapes, dtypes): + if jtu.device_under_test() == "cpu": + raise unittest.SkipTest("TODO(b/280418378) test fails after LLVM update") rng = jtu.rand_default(self.rng()) scipy_fun = osp_stats.truncnorm.pdf lax_fun = lsp_stats.truncnorm.pdf