Fixed margin bug

rosalindfranklininstitute · May 14, 2024 · 7a1c222 · 7a1c222
1 parent 33f6b2f
commit 7a1c222
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/parakeet/simulate/engine.py b/src/parakeet/simulate/engine.py
@@ -404,7 +404,11 @@ def callback(z0, z1, V):
                 % (z0, z1, index)
             )
             if index < out.data.shape[0]:
-                out.data[index, :, :] = V[margin:-margin, margin:-margin].T
+                x0 = margin
+                y0 = margin
+                x1 = V.shape[1] - margin
+                y1 = V.shape[0] - margin
+                out.data[index, :, :] = V[y0:y1, x0:x1].T
 
         # Run the simulation
         if masker is not None: