提交 9b787864 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

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

上级 73f37330
......@@ -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",
[
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论