Fixed patch slices generation in PatchSampler to always retrieve patc…

…hes of the defined shape
TheJacksonLaboratory · May 2, 2024 · aaa51f6 · aaa51f6
1 parent b894985
commit aaa51f6
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/zarrdataset/_samplers.py b/zarrdataset/_samplers.py
@@ -320,9 +320,7 @@ def _compute_toplefts_slices(self, mask: ImageBase, image_shape: dict,
                           + tls[mask.axes.index(ax)] + patch_size[ax])
 
                     curr_tl.append((ax, slice(tl - pad[ax],
-                                              (br if br <= image_shape[ax]
-                                               else image_shape[ax])
-                                              + pad[ax])))
+                                              br + pad[ax])))
 
                 else:
                     curr_tl.append((ax, slice(0, 1)))