Add an xfail for newer JAX versions that change sampler size behavior

aesara-devs · Dec 15, 2021 · d648969 · d648969
1 parent e4ffb02
commit d648969
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/link/test_jax.py b/tests/link/test_jax.py
@@ -1207,6 +1207,10 @@ def test_extra_ops_omni():
     compare_jax_and_py(fgraph, [])
 
 
+@pytest.mark.xfail(
+    version_parse(jax.__version__) >= version_parse("0.2.26"),
+    reason="JAX samplers require concrete/static shape values?",
+)
 @pytest.mark.parametrize(
     "at_dist, dist_params, rng, size",
     [