Skip symmetric product test on older jaxlibs.

The new symmetric product operator will appear to jaxlib 0.4.32. PiperOrigin-RevId: 672311569
jax-ml · Sep 8, 2024 · 5af1efb · 5af1efb
1 parent 79dabe5
commit 5af1efb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/linalg_test.py b/tests/linalg_test.py
@@ -16,6 +16,7 @@
 
 from functools import partial
 import itertools
+import unittest
 
 import numpy as np
 import scipy
@@ -2193,6 +2194,9 @@ def testHilbert(self, n):
       symmetrize_output=[True, False],
   )
   @jtu.skip_on_devices("tpu")
+  @unittest.skipIf(
+      jax._src.lib.version < (0, 4, 32), "requires jaxlib >= 0.4.32"
+  )
   def testSymmetricProduct(self, shape, dtype, symmetrize_output):
     rng = jtu.rand_default(self.rng())
     batch_size = 10