Skip JAX test if numpyro is not installed

jessegrabowski · jessegrabowski · commit 020e246c7523 · 2024-06-28T23:02:05.000+08:00
diff --git a/pymc_experimental/tests/statespace/test_statespace_JAX.py b/pymc_experimental/tests/statespace/test_statespace_JAX.py
@@ -25,6 +25,7 @@
 )
 
 pytest.importorskip("jax")
+pytest.importorskip("numpyro")
 
 
 floatX = pytensor.config.floatX

Original file line number	Diff line number	Diff line change
`@@ -25,6 +25,7 @@`
`25`	`25`	`)`
`26`	`26`
`27`	`27`	`pytest.importorskip("jax")`
	`28`	`+pytest.importorskip("numpyro")`
`28`	`29`
`29`	`30`
`30`	`31`	`floatX = pytensor.config.floatX`