Avoid mismatch with declared linop dtype

lanl · Oct 18, 2024 · 38ba3ae · 38ba3ae
1 parent c4a7a49
commit 38ba3ae
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scico/linop/xray/_xray.py b/scico/linop/xray/_xray.py
@@ -273,7 +273,7 @@ def _back_project(
             y[jnp.arange(len(angles)).reshape(-1, 1, 1), inds + 1] * weights_valid, axis=0
         )
 
-        return HTy
+        return HTy.astype(jnp.float32)
 
     @staticmethod
     @partial(jax.jit, static_argnames=["nx"])

diff --git a/scico/test/flax/test_inv.py b/scico/test/flax/test_inv.py
@@ -155,7 +155,7 @@ def setup_method(self, method):
         xt, key = random.randn((2 * self.bsize, self.N, self.N, self.chn), seed=4321)
 
         self.nproj = 60  # number of projections
-        angles = np.linspace(0, np.pi, self.nproj)  # evenly spaced projection angles
+        angles = np.linspace(0, np.pi, self.nproj, endpoint=False, dtype=np.float32)
         self.opCT = XRayTransform2D(
             input_shape=(self.N, self.N),
             det_count=self.N,