Skip to content

Commit 757a10c

Browse files
committed
Skip failing JAX test
Started failing in 0.4.36: jax-ml/jax#26888 Skip failing JAX test Started failing in 0.4.36 and seems to be fixed in 0.5.1
1 parent 6a19a37 commit 757a10c

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

tests/link/jax/test_pad.py

+9-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import numpy as np
22
import pytest
3+
from packaging import version
34

45
import pytensor.tensor as pt
56
from pytensor import config
@@ -16,7 +17,14 @@
1617
"mode, kwargs",
1718
[
1819
("constant", {"constant_values": 0}),
19-
("constant", {"constant_values": (1, 2)}),
20+
pytest.param(
21+
"constant",
22+
{"constant_values": (1, 2)},
23+
marks=pytest.mark.skipif(
24+
version.parse(jax.__version__) > version.parse("0.4.35"),
25+
reason="Bug in JAX: https://github.com/jax-ml/jax/issues/26888",
26+
),
27+
),
2028
("edge", {}),
2129
("linear_ramp", {"end_values": 0}),
2230
("linear_ramp", {"end_values": (1, 2)}),

0 commit comments

Comments
 (0)