diff --git a/xlb/operator/boundary_masker/indices_boundary_masker.py b/xlb/operator/boundary_masker/indices_boundary_masker.py index 36f9a36..0c1f7e1 100644 --- a/xlb/operator/boundary_masker/indices_boundary_masker.py +++ b/xlb/operator/boundary_masker/indices_boundary_masker.py @@ -93,7 +93,6 @@ def jax_implementation(self, bclist, bc_mask, missing_mask, start_index=None): bc_mask = bc_mask.at[0].set(bmap[pad_x:-pad_x, pad_y:-pad_y, pad_z:-pad_z]) return bc_mask, missing_mask - def _construct_warp(self): # Make constants for warp _c = self.velocity_set.c