Resolve errors in jitting method

lanl · Oct 6, 2024 · ff1e235 · ff1e235
1 parent 97f3b05
commit ff1e235
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scico/linop/xray/_xray.py b/scico/linop/xray/_xray.py
@@ -141,8 +141,8 @@ def fbp(self, y: ArrayLike) -> snp.Array:
         """
 
         N = y.shape[1]
-        nvec = np.arange(N) - (N - 1) // 2
-        dx = np.sqrt(self.dx[0] * self.dx[1])  # type: ignore
+        nvec = snp.arange(N) - (N - 1) // 2
+        dx = snp.sqrt(self.dx[0] * self.dx[1])  # type: ignore
         h = XRayTransform2D._ramp_filter(nvec, 1.0 / dx)
 
         # Apply ramp filter in the frequency domain, padding to avoid