fix

Jammy2211 · Dec 18, 2023 · 3b4e467 · 3b4e467
1 parent dcbb09b
commit 3b4e467
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/autoarray/structures/structure_decorators.py b/autoarray/structures/structure_decorators.py
@@ -577,9 +577,10 @@ def wrapper(
             grid_radial_scale = np.where(
                 grid_radii < grid_radial_minimum, grid_radial_minimum / grid_radii, 1.0
             )
-            moved_grid = grid.with_new_array(
-                np.multiply(grid, grid_radial_scale[:, None])
-            )
+            moved_grid = np.multiply(grid, grid_radial_scale[:, None])
+
+            if hasattr(grid, "with_new_array"):
+                moved_grid = grid.with_new_array(moved_grid)
 
         moved_grid[np.isnan(np.array(moved_grid))] = grid_radial_minimum