Skip to content
This repository was archived by the owner on Nov 17, 2025. It is now read-only.

Commit bd697dc

Browse files
brandonwillardrlouf
authored andcommitted
Use separate DB queries for each JAX test mode
1 parent 0c06f92 commit bd697dc

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tests/link/jax/test_basic.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,12 @@ def set_aesara_flags():
2727
jax = pytest.importorskip("jax")
2828

2929

30-
opts = RewriteDatabaseQuery(include=["jax"], exclude=["cxx_only", "BlasOpt"])
31-
jax_mode = Mode(JAXLinker(), opts)
32-
py_mode = Mode("py", opts)
30+
jax_mode = Mode(
31+
JAXLinker(), RewriteDatabaseQuery(include=["jax"], exclude=["cxx_only", "BlasOpt"])
32+
)
33+
py_mode = Mode(
34+
"py", RewriteDatabaseQuery(include=[None], exclude=["cxx_only", "BlasOpt"])
35+
)
3336

3437

3538
def compare_jax_and_py(

0 commit comments

Comments
 (0)