Skip to content

Commit a0d9ecf

Browse files
committed
Skip failing JAX test
Started failing in 0.4.36 and seems to be fixed in 0.5.1
1 parent cd649ab commit a0d9ecf

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

tests/link/jax/test_pad.py

Lines changed: 8 additions & 1 deletion
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,13 @@
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.skipif(
24+
version.parse(jax.__version__) < version.parse("0.5.1")
25+
),
26+
),
2027
("edge", {}),
2128
("linear_ramp", {"end_values": 0}),
2229
("linear_ramp", {"end_values": (1, 2)}),

0 commit comments

Comments
 (0)