We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cd649ab commit 8ad0812Copy full SHA for 8ad0812
tests/link/jax/test_pad.py
@@ -1,5 +1,6 @@
1
import numpy as np
2
import pytest
3
+from packaging import version
4
5
import pytensor.tensor as pt
6
from pytensor import config
@@ -16,7 +17,13 @@
16
17
"mode, kwargs",
18
[
19
("constant", {"constant_values": 0}),
- ("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.5.1")
25
+ ),
26
27
("edge", {}),
28
("linear_ramp", {"end_values": 0}),
29
("linear_ramp", {"end_values": (1, 2)}),
0 commit comments