Skip to content

Commit 783ff84

Browse files
Jammy2211Jammy2211
authored andcommitted
remove numba from two more functions ing rid_2d_util
1 parent 3832bca commit 783ff84

1 file changed

Lines changed: 5 additions & 6 deletions

File tree

autoarray/structures/grids/grid_2d_util.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -395,8 +395,6 @@ def grid_2d_via_shape_native_from(
395395
origin=origin,
396396
)
397397

398-
399-
@numba_util.jit()
400398
def _radial_projected_shape_slim_from(
401399
extent: np.ndarray,
402400
centre: Tuple[float, float],
@@ -471,7 +469,6 @@ def _radial_projected_shape_slim_from(
471469
return int((scaled_distance / pixel_scale)) + 1
472470

473471

474-
@numba_util.jit()
475472
def grid_scaled_2d_slim_radial_projected_from(
476473
extent: np.ndarray,
477474
centre: Tuple[float, float],
@@ -562,9 +559,11 @@ def grid_scaled_2d_slim_radial_projected_from(
562559

563560
radii = centre[1]
564561

565-
for slim_index in range(shape_slim):
566-
grid_scaled_2d_slim_radii[slim_index, 1] = radii
567-
radii += pixel_scale
562+
# Create an array of radii values spaced by pixel_scale
563+
radii_array = radii + pixel_scale * np.arange(shape_slim)
564+
565+
# Assign all values at once to the second column (index 1)
566+
grid_scaled_2d_slim_radii[:, 1] = radii_array
568567

569568
return grid_scaled_2d_slim_radii + 1e-6
570569

0 commit comments

Comments
 (0)