Fix compatibility issue with older jax versions

lanl · Oct 31, 2023 · 1a2f2d4 · 1a2f2d4
1 parent 7b3ecfa
commit 1a2f2d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/scripts/ct_projector_comparison.py b/examples/scripts/ct_projector_comparison.py
@@ -133,7 +133,7 @@
 
 y = np.zeros(H.output_shape, dtype=np.float32)
 y[num_angles // 3, det_count // 2] = 1.0
-y = jax.array(y)
+y = jnp.array(y)
 
 HTys = {}
 for name, H in projectors.items():