Disable linalg_test.testCond.

Issue: 2203 This test was added in jax-ml#2125 but is failing in internal tests.
belvo · Feb 10, 2020 · e81024f · e81024f
1 parent 0b31342
commit e81024f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/linalg_test.py b/tests/linalg_test.py
@@ -584,6 +584,8 @@ def testQrBatching(self, shape, dtype, rng_factory):
       for shape in [(1, 1), (4, 4), (2, 3, 5), (5, 5, 5), (20, 20), (5, 10)]
       for pnorm in [np.inf, -np.inf, 1, -1, 2, -2, 'fro']
       for dtype in float_types + complex_types))
+  @jtu.skip_on_devices("tpu")  # SVD is not implemented on the TPU backend
+  @jtu.skip_on_devices("gpu")  # TODO(#2203): numerical errors
   def testCond(self, shape, pnorm, dtype):
     _skip_if_unsupported_type(dtype)