remove where

Author: Jammy2211

autoarray/inversion/pixelization/border_relocator.py

@@ -311,9 +311,10 @@ def relocated_grid_via_ellipse_border_from(grid, origin, a, b, phi, xp=np, eps=1
 
     moved = xp.stack([origin[0] + dy2, origin[1] + dx2], axis=1)
 
-    return moved
+#    return xp.where(outside[:, None], moved, grid)
+
+    return grid + (moved - grid) * outside.astype(grid.dtype)[:, None]
 
-    return xp.where(outside[:, None], moved, grid)
 
 
 class BorderRelocator: