Skip to content

Commit b4c0db4

Browse files
Jammy2211Jammy2211
authored andcommitted
remove many np.array() conversions
1 parent e8d767a commit b4c0db4

11 files changed

Lines changed: 26 additions & 582 deletions

File tree

autoarray/fit/fit_interferometer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ def noise_normalization(self) -> float:
122122
[Noise_Term] = sum(log(2*pi*[Noise]**2.0))
123123
"""
124124
return fit_util.noise_normalization_complex_from(
125-
noise_map=self.noise_map,
125+
noise_map=self.noise_map.array,
126126
)
127127

128128
@property

autoarray/geometry/geometry_2d.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -294,7 +294,7 @@ def grid_pixel_indexes_2d_from(self, grid_scaled_2d: Grid2D) -> Array2D:
294294
from autoarray.structures.arrays.uniform_2d import Array2D
295295

296296
grid_pixel_indexes_2d = geometry_util.grid_pixel_indexes_2d_slim_from(
297-
grid_scaled_2d_slim=np.array(grid_scaled_2d),
297+
grid_scaled_2d_slim=grid_scaled_2d,
298298
shape_native=self.shape_native,
299299
pixel_scales=self.pixel_scales,
300300
origin=self.origin,
@@ -320,7 +320,7 @@ def grid_scaled_2d_from(self, grid_pixels_2d: Grid2D) -> Grid2D:
320320
from autoarray.structures.grids.uniform_2d import Grid2D
321321

322322
grid_scaled_1d = geometry_util.grid_scaled_2d_slim_from(
323-
grid_pixels_2d_slim=np.array(grid_pixels_2d),
323+
grid_pixels_2d_slim=grid_pixels_2d,
324324
shape_native=self.shape_native,
325325
pixel_scales=self.pixel_scales,
326326
origin=self.origin,

autoarray/inversion/pixelization/image_mesh/kmeans.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ def image_plane_mesh_grid_from(
105105
max_iter=5,
106106
)
107107

108-
grid = mask.derive_grid.unmasked
108+
grid = mask.derive_grid.unmasked.array
109109

110110
try:
111111
kmeans = kmeans.fit(X=grid, sample_weight=weight_map)

autoarray/mask/derive/grid_2d.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def all_false(self) -> Grid2D:
112112
origin=self.mask.origin,
113113
)
114114

115-
return Grid2D(values=np.array(grid_slim), mask=self.mask.derive_mask.all_false)
115+
return Grid2D(values=grid_slim, mask=self.mask.derive_mask.all_false)
116116

117117
@property
118118
def unmasked(self) -> Grid2D:
@@ -163,7 +163,7 @@ def unmasked(self) -> Grid2D:
163163
pixel_scales=self.mask.pixel_scales,
164164
origin=self.mask.origin,
165165
)
166-
return Grid2D(values=np.array(grid_2d), mask=self.mask)
166+
return Grid2D(values=grid_2d, mask=self.mask)
167167

168168
@property
169169
def edge(self) -> Grid2D:

autoarray/mask/derive/indexes_2d.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def unmasked_slim(self) -> np.ndarray:
112112
print(derive_indexes_2d.unmasked_slim)
113113
"""
114114
return mask_2d_util.mask_slim_indexes_from(
115-
mask_2d=np.array(self.mask), return_masked_indexes=False
115+
mask_2d=self.mask, return_masked_indexes=False
116116
).astype("int")
117117

118118
@property
@@ -154,7 +154,7 @@ def masked_slim(self) -> np.ndarray:
154154
print(derive_indexes_2d.masked_slim)
155155
"""
156156
return mask_2d_util.mask_slim_indexes_from(
157-
mask_2d=np.array(self.mask), return_masked_indexes=True
157+
mask_2d=self.mask, return_masked_indexes=True
158158
).astype("int")
159159

160160
@property
@@ -202,7 +202,7 @@ def edge_slim(self) -> np.ndarray:
202202
print(derive_indexes_2d.edge_slim)
203203
"""
204204
return mask_2d_util.edge_1d_indexes_from(
205-
mask_2d=np.array(self.mask).astype("bool")
205+
mask_2d=self.mask.astype("bool")
206206
).astype("int")
207207

208208
@property
@@ -304,7 +304,7 @@ def border_slim(self) -> np.ndarray:
304304
print(derive_indexes_2d.border_slim)
305305
"""
306306
return mask_2d_util.border_slim_indexes_from(
307-
mask_2d=np.array(self.mask).astype("bool")
307+
mask_2d=self.mask.astype("bool")
308308
).astype("int")
309309

310310
@property
@@ -409,5 +409,5 @@ def native_for_slim(self) -> np.ndarray:
409409
print(derive_indexes_2d.native_for_slim)
410410
"""
411411
return mask_2d_util.native_index_for_slim_index_2d_from(
412-
mask_2d=np.array(self.mask),
412+
mask_2d=self.mask,
413413
).astype("int")

autoarray/mask/derive/mask_2d.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,7 @@ def blurring_from(self, kernel_shape_native: Tuple[int, int]) -> Mask2D:
204204
raise exc.MaskException("psf_size of exterior region must be odd")
205205

206206
blurring_mask = mask_2d_util.blurring_mask_2d_from(
207-
mask_2d=np.array(self.mask),
207+
mask_2d=self.mask,
208208
kernel_shape_native=kernel_shape_native,
209209
)
210210

@@ -325,7 +325,7 @@ def edge_buffed(self) -> Mask2D:
325325
from autoarray.mask.mask_2d import Mask2D
326326

327327
edge_buffed_mask = mask_2d_util.buffed_mask_2d_from(
328-
mask_2d=np.array(self.mask)
328+
mask_2d=self.mask
329329
).astype("bool")
330330

331331
return Mask2D(

autoarray/mask/mask_1d.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -153,9 +153,7 @@ def from_fits(
153153
"""
154154

155155
return cls(
156-
mask=np.array(
157-
array_1d_util.numpy_array_1d_via_fits_from(file_path=file_path, hdu=hdu)
158-
),
156+
mask=array_1d_util.numpy_array_1d_via_fits_from(file_path=file_path, hdu=hdu),
159157
pixel_scales=pixel_scales,
160158
origin=origin,
161159
)

autoarray/mask/mask_2d.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -331,7 +331,7 @@ def circular(
331331
)
332332

333333
return cls(
334-
mask=np.array(mask),
334+
mask=mask,
335335
pixel_scales=pixel_scales,
336336
origin=origin,
337337
invert=invert,
@@ -385,7 +385,7 @@ def circular_annular(
385385
)
386386

387387
return cls(
388-
mask=np.array(mask),
388+
mask=mask,
389389
pixel_scales=pixel_scales,
390390
origin=origin,
391391
invert=invert,
@@ -442,7 +442,7 @@ def elliptical(
442442
)
443443

444444
return cls(
445-
mask=np.array(mask),
445+
mask=mask,
446446
pixel_scales=pixel_scales,
447447
origin=origin,
448448
invert=invert,
@@ -512,7 +512,7 @@ def elliptical_annular(
512512
)
513513

514514
return cls(
515-
mask=np.array(mask),
515+
mask=mask,
516516
pixel_scales=pixel_scales,
517517
origin=origin,
518518
invert=invert,
@@ -560,7 +560,7 @@ def from_pixel_coordinates(
560560
)
561561

562562
return cls(
563-
mask=np.array(mask),
563+
mask=mask,
564564
pixel_scales=pixel_scales,
565565
origin=origin,
566566
invert=invert,
@@ -598,7 +598,7 @@ def from_fits(
598598
mask = np.invert(mask.astype("bool"))
599599

600600
mask = Mask2D(
601-
mask=np.array(mask),
601+
mask=mask,
602602
pixel_scales=pixel_scales,
603603
origin=origin,
604604
)
@@ -680,7 +680,7 @@ def header_dict(self) -> Dict:
680680
@property
681681
def mask_centre(self) -> Tuple[float, float]:
682682
grid = grid_2d_util.grid_2d_slim_via_mask_from(
683-
mask_2d=np.array(self),
683+
mask_2d=self,
684684
pixel_scales=self.pixel_scales,
685685
origin=self.origin,
686686
)
@@ -697,7 +697,7 @@ def shape_native_masked_pixels(self) -> Tuple[int, int]:
697697
and 12 False entries going horizontally in the central regions of the mask, then shape_masked_pixels=(15,12).
698698
"""
699699

700-
where = np.array(np.where(np.invert(self.astype("bool"))))
700+
where = np.where(np.invert(self.astype("bool")))
701701
y0, x0 = np.amin(where, axis=1)
702702
y1, x1 = np.amax(where, axis=1)
703703

autoarray/operators/transformer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ def visibilities_from(self, image: Array2D) -> Visibilities:
140140
else:
141141
visibilities = transformer_util.visibilities_from(
142142
image_1d=image.slim.array,
143-
grid_radians=np.array(self.grid),
143+
grid_radians=self.grid.array,
144144
uv_wavelengths=self.uv_wavelengths,
145145
)
146146

@@ -170,7 +170,7 @@ def image_from(
170170
"""
171171
image_slim = transformer_util.image_direct_from(
172172
visibilities=visibilities.in_array,
173-
grid_radians=np.array(self.grid.array),
173+
grid_radians=self.grid.array,
174174
uv_wavelengths=self.uv_wavelengths,
175175
)
176176

@@ -210,7 +210,7 @@ def transform_mapping_matrix(self, mapping_matrix: np.ndarray) -> np.ndarray:
210210

211211
return transformer_util.transformed_mapping_matrix_from(
212212
mapping_matrix=mapping_matrix,
213-
grid_radians=np.array(self.grid),
213+
grid_radians=self.grid.array,
214214
uv_wavelengths=self.uv_wavelengths,
215215
)
216216

autoarray/plot/visuals/two_d.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def plot_via_plotter(self, plotter, grid_indexes=None, mapper=None, geometry=Non
5656

5757
if self.mask is not None:
5858
plotter.mask_scatter.scatter_grid(
59-
grid=np.array(self.mask.derive_grid.edge.array)
59+
grid=self.mask.derive_grid.edge.array
6060
)
6161

6262
if self.border is not None:

0 commit comments

Comments
 (0)