From 019c296e59ded3579edbb597d4139c9ed7617a79 Mon Sep 17 00:00:00 2001
From: Stephanie Ribet <stephanie.ribet@gmail.com>
Date: Fri, 30 Aug 2024 11:32:19 -0700
Subject: [PATCH] left in hard coded DP shape

---
 py4DSTEM/tomography/tomography.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/py4DSTEM/tomography/tomography.py b/py4DSTEM/tomography/tomography.py
index f53e8d32f..1ced37074 100644
--- 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(