black

Author: Jammy2211

autoarray/dataset/imaging/dataset.py

@@ -214,6 +214,17 @@ def w_tilde(self):
 
         logger.info("IMAGING - Computing W-Tilde... May take a moment.")
 
+        try:
+            import numba
+        except ModuleNotFoundError:
+            raise exc.InversionException(
+                "Inversion w-tilde functionality (pixelized reconstructions) is "
+                "disabled if numba is not installed.\n\n"
+                "This is because the run-times without numba are too slow.\n\n"
+                "Please install numba, which is described at the following web page:\n\n"
+                "https://pyautolens.readthedocs.io/en/latest/installation/overview.html"
+            )
+
         (
             curvature_preload,
             indexes,

autoarray/dataset/imaging/w_tilde.py

@@ -86,8 +86,8 @@ def w_matrix(self):
         """
 
         return inversion_imaging_numba_util.w_tilde_curvature_imaging_from(
-            noise_map_native=np.array(self.noise_map.native.array).astype("float64"),
-            kernel_native=np.array(self.psf.native.array).astype("float64"),
+            noise_map_native=self.noise_map.native.array,
+            kernel_native=self.psf.native.array,
             native_index_for_slim_index=np.array(
                 self.mask.derive_indexes.native_for_slim
             ).astype("int"),
@@ -97,7 +97,7 @@ def w_matrix(self):
     def psf_operator_matrix_dense(self):
 
         return inversion_imaging_util.psf_operator_matrix_dense_from(
-            kernel_native=np.array(self.psf.native.array).astype("float64"),
+            kernel_native=self.psf.native.array,
             native_index_for_slim_index=np.array(
                 self.mask.derive_indexes.native_for_slim
             ).astype("int"),

autoarray/dataset/interferometer/dataset.py

@@ -186,6 +186,17 @@ def w_tilde(self):
 
         logger.info("INTERFEROMETER - Computing W-Tilde... May take a moment.")
 
+        try:
+            import numba
+        except ModuleNotFoundError:
+            raise exc.InversionException(
+                "Inversion w-tilde functionality (pixelized reconstructions) is "
+                "disabled if numba is not installed.\n\n"
+                "This is because the run-times without numba are too slow.\n\n"
+                "Please install numba, which is described at the following web page:\n\n"
+                "https://pyautolens.readthedocs.io/en/latest/installation/overview.html"
+            )
+
         curvature_preload = (
             inversion_interferometer_util.w_tilde_curvature_preload_interferometer_from(
                 noise_map_real=np.array(self.noise_map.real),

autoarray/inversion/inversion/imaging/w_tilde.py

@@ -515,7 +515,7 @@ def mapped_reconstructed_data_dict(self) -> Dict[LinearObj, Array2D]:
                     data_to_pix_unique=linear_obj.unique_mappings.data_to_pix_unique,
                     data_weights=linear_obj.unique_mappings.data_weights,
                     pix_lengths=linear_obj.unique_mappings.pix_lengths,
-                    reconstruction=reconstruction,
+                    reconstruction=np.array(reconstruction),
                 )
 
                 mapped_reconstructed_image = self.psf.convolve_image_no_blurring(

autoarray/inversion/inversion/settings.py

@@ -10,7 +10,7 @@
 class SettingsInversion:
     def __init__(
         self,
-        use_w_tilde: bool = False,
+        use_w_tilde: bool = True,
         use_positive_only_solver: Optional[bool] = None,
         positive_only_uses_p_initial: Optional[bool] = None,
         use_border_relocator: Optional[bool] = None,

autoarray/inversion/pixelization/mappers/delaunay.py

@@ -168,7 +168,7 @@ def pix_sub_weights_split_cross(self) -> PixSubWeights:
 
         splitted_weights = mapper_numba_util.pixel_weights_delaunay_from(
             source_plane_data_grid=self.source_plane_mesh_grid.split_cross,
-            source_plane_mesh_grid=np.array(self.source_plane_mesh_grid),
+            source_plane_mesh_grid=np.array(self.source_plane_mesh_grid.array),
             slim_index_for_sub_slim_index=self.source_plane_mesh_grid.split_cross,
             pix_indexes_for_sub_slim_index=splitted_mappings.astype("int"),
         )

autoarray/inversion/pixelization/mappers/voronoi.py

@@ -126,8 +126,8 @@ def pix_sub_weights(self) -> PixSubWeights:
         """
 
         mappings, sizes, weights = mapper_numba_util.pix_size_weights_voronoi_nn_from(
-            grid=self.source_plane_data_grid.over_sampled,
-            mesh_grid=self.source_plane_mesh_grid,
+            grid=np.array(self.source_plane_data_grid.over_sampled),
+            mesh_grid=np.array(self.source_plane_mesh_grid.array),
         )
 
         mappings = mappings.astype("int")

autoarray/operators/transformer.py

@@ -383,7 +383,7 @@ def visibilities_from(self, image: Array2D) -> Visibilities:
 
         return Visibilities(
             visibilities=self.forward(
-                image.native[::-1, :]
+                image.native.array[::-1, :]
             )  # flip due to PyNUFFT internal flip
         )
 

autoarray/plot/visuals/two_d.py

@@ -1,4 +1,5 @@
 from matplotlib import patches as ptch
+import numpy as np
 from typing import List, Optional, Union
 
 from autoarray.mask.mask_2d import Mask2D
@@ -91,6 +92,6 @@ def plot_via_plotter(self, plotter, grid_indexes=None):
         if self.indexes is not None and grid_indexes is not None:
 
             plotter.index_scatter.scatter_grid_indexes(
-                grid=grid_indexes,
+                grid=np.array(grid_indexes),
                 indexes=self.indexes,
             )

autoarray/plot/wrap/two_d/delaunay_drawer.py

@@ -104,8 +104,8 @@ def draw_delaunay_pixels(
                 colorbar_tickparams.set(cb=cb)
 
         ax.tripcolor(
-            source_pixelization_grid[:, 1],
-            source_pixelization_grid[:, 0],
+            source_pixelization_grid.array[:, 1],
+            source_pixelization_grid.array[:, 0],
             simplices,
             facecolors=facecolors,
             edgecolors="None",