Resolve doctest failure of mysterious origin

lanl · May 14, 2024 · 63ca60e · 63ca60e
1 parent ce0b473
commit 63ca60e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/conftest.py b/conftest.py
@@ -14,6 +14,8 @@
     have_ray = True
     ray.init(num_cpus=1)  # call required to be here: see ray-project/ray#44087
 
+import jax.numpy as jnp
+
 import scico.numpy as snp
 
 
@@ -38,3 +40,4 @@ def add_modules(doctest_namespace):
     """
     doctest_namespace["np"] = np
     doctest_namespace["snp"] = snp
+    doctest_namespace["jnp"] = jnp