left in hard coded DP shape

py4dstem · Aug 30, 2024 · 019c296 · 019c296
1 parent ac9364a
commit 019c296
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py4DSTEM/tomography/tomography.py b/py4DSTEM/tomography/tomography.py
@@ -1024,8 +1024,8 @@ def _forward(
         dy_diff = line_y_diff - yF_diff
         dz_diff = line_z_diff - zF_diff
 
-        qx = xp.arange(11)
-        qy = xp.arange(11)
+        qx = xp.arange(s[-1])
+        qy = xp.arange(s[-1])
         qxx, qyy = xp.meshgrid(qx, qy)
 
         ind0_diff = np.hstack(