Overlooked api updates

lanl · Jan 19, 2024 · 5800df6 · 5800df6
1 parent 3afbab2
commit 5800df6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/examples/scripts/ct_projector_comparison.py b/examples/scripts/ct_projector_comparison.py
@@ -52,8 +52,8 @@
 timer.stop("scico_init")
 
 timer.start("astra_init")
-projectors["astra"] = astra.XRayTransform(
-    (N, N), detector_spacing=1.0, det_count=det_count, angles=angles - jnp.pi / 2.0
+projectors["astra"] = astra.XRayTransform2D(
+    (N, N), det_count=det_count, det_spacing=1.0, angles=angles - jnp.pi / 2.0
 )
 timer.stop("astra_init")
 

diff --git a/scico/test/flax/test_inv.py b/scico/test/flax/test_inv.py
@@ -16,7 +16,7 @@
 from scico.linop import CircularConvolve, Identity
 
 if have_astra:
-    from scico.linop.xray.astra import XRayTransform
+    from scico.linop.xray.astra import XRayTransform2D
 
 
 os.environ["XLA_FLAGS"] = "--xla_force_host_platform_device_count=8"
@@ -163,10 +163,10 @@ def setup_method(self, method):
 
         self.nproj = 60  # number of projections
         angles = np.linspace(0, np.pi, self.nproj)  # evenly spaced projection angles
-        self.opCT = XRayTransform(
+        self.opCT = XRayTransform2D(
             input_shape=(self.N, self.N),
-            detector_spacing=1,
             det_count=self.N,
+            det_spacing=1.0,
             angles=angles,
         )  # Radon transform operator
         a_f = lambda v: jnp.atleast_3d(self.opCT(v.squeeze()))