quirks removed

Author: Jammy2211

autogalaxy/ellipse/model/plotter_interface.py

@@ -46,8 +46,8 @@ def should_plot(name):
             dataset_plotter.subplot_dataset()
 
         image_list = [
-            dataset.data.native_for_fits,
-            dataset.noise_map.native_for_fits,
+            dataset.data.native,
+            dataset.noise_map.native,
         ]
 
         hdu_list = hdu_list_for_output_from(

autogalaxy/galaxy/galaxies.py

@@ -172,9 +172,7 @@ def deflections_yx_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> np.ndarr
         grid
             The 2D (y, x) coordinates where values of the deflections are evaluated.
         """
-        if self:
-            return sum(map(lambda g: g.deflections_yx_2d_from(grid=grid), self))
-        return np.zeros(shape=(grid.shape[0], 2))
+        return sum(map(lambda g: g.deflections_yx_2d_from(grid=grid), self))
 
     @aa.grid_dec.to_grid
     def traced_grid_2d_from(self, grid: aa.type.Grid2DLike) -> aa.type.Grid2DLike:
@@ -205,9 +203,7 @@ def convergence_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> np.ndarray:
         grid
             The 2D (y, x) coordinates where values of the convergence are evaluated.
         """
-        if self:
-            return sum(map(lambda g: g.convergence_2d_from(grid=grid), self))
-        return np.zeros((grid.shape[0],))
+        return sum(map(lambda g: g.convergence_2d_from(grid=grid), self))
 
     @aa.grid_dec.to_array
     def potential_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> np.ndarray:
@@ -231,9 +227,7 @@ def potential_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> np.ndarray:
         grid
             The 2D (y, x) coordinates where values of the potential are evaluated.
         """
-        if self:
-            return sum(map(lambda g: g.potential_2d_from(grid=grid), self))
-        return np.zeros((grid.shape[0],))
+        return sum(map(lambda g: g.potential_2d_from(grid=grid), self))
 
     def has(self, cls: Union[Type, Tuple[Type]]) -> bool:
         """

autogalaxy/galaxy/galaxy.py

@@ -229,10 +229,7 @@ def image_2d_from(
             > 0
         ):
             return sum(self.image_2d_list_from(grid=grid, operated_only=operated_only))
-
-        return np.zeros((grid.shape[0],))
-
-    #        return jnp.zeros((grid.shape[0],))
+        return jnp.zeros((grid.shape[0],))
 
     @aa.grid_dec.to_projected
     def image_1d_from(self, grid: aa.type.Grid2DLike) -> np.ndarray:
@@ -264,9 +261,8 @@ def image_1d_from(self, grid: aa.type.Grid2DLike) -> np.ndarray:
                 image_1d_list.append(light_profile.image_1d_from(grid=grid_radial))
 
             return sum(image_1d_list)
-        return np.zeros((grid.shape[0],))
 
-    #   return jnp.zeros((grid.shape[0],))
+        return jnp.zeros((grid.shape[0],))
 
     @aa.grid_dec.to_vector_yx
     def deflections_yx_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> np.ndarray:
@@ -294,9 +290,8 @@ def deflections_yx_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> np.ndarr
                     self.cls_list_from(cls=MassProfile),
                 )
             )
-        return np.zeros((grid.shape[0], 2))
 
-    #        return jnp.zeros((grid.shape[0], 2))
+        return jnp.zeros((grid.shape[0], 2))
 
     @aa.grid_dec.to_array
     def convergence_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> np.ndarray:
@@ -324,9 +319,7 @@ def convergence_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> np.ndarray:
                 )
             )
 
-        return np.zeros((grid.shape[0],))
-
-    #        return jnp.zeros((grid.shape[0],))
+        return jnp.zeros((grid.shape[0],))
 
     @aa.grid_dec.to_grid
     def traced_grid_2d_from(self, grid: aa.type.Grid2DLike) -> aa.type.Grid2DLike:
@@ -376,9 +369,7 @@ def convergence_1d_from(self, grid: aa.type.Grid1D2DLike) -> np.ndarray:
 
             return sum(convergence_1d_list)
 
-        return np.zeros((grid.shape[0],))
-
-    #     return jnp.zeros((grid.shape[0],))
+        return jnp.zeros((grid.shape[0],))
 
     @aa.grid_dec.to_array
     def potential_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> np.ndarray:
@@ -405,9 +396,7 @@ def potential_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> np.ndarray:
                     self.cls_list_from(cls=MassProfile),
                 )
             )
-        return np.zeros((grid.shape[0],))
-
-    #        return jnp.zeros((grid.shape[0],))
+        return jnp.zeros((grid.shape[0],))
 
     @aa.grid_dec.to_projected
     def potential_1d_from(self, grid: aa.type.Grid2DLike) -> np.ndarray:
@@ -439,9 +428,8 @@ def potential_1d_from(self, grid: aa.type.Grid2DLike) -> np.ndarray:
                 )
 
             return sum(potential_1d_list)
-        return np.zeros((grid.shape[0],))
 
-    #        return jnp.zeros((grid.shape[0],))
+        return jnp.zeros((grid.shape[0],))
 
     @property
     def half_light_radius(self):

autogalaxy/profiles/basis.py

@@ -1,3 +1,4 @@
+import jax.numpy as jnp
 import numpy as np
 from typing import Dict, List, Optional, Union
 
@@ -128,7 +129,7 @@ def image_2d_list_from(
             (
                 light_profile.image_2d_from(grid=grid, operated_only=operated_only)
                 if not isinstance(light_profile, lp_linear.LightProfileLinear)
-                else np.zeros((grid.shape[0],))
+                else jnp.zeros((grid.shape[0],))
             )
             for light_profile in self.light_profile_list
         ]
@@ -154,7 +155,7 @@ def convergence_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> aa.Array2D:
             return sum(
                 [mass.convergence_2d_from(grid=grid) for mass in self.profile_list]
             )
-        return np.zeros((grid.shape[0],))
+        return jnp.zeros((grid.shape[0],))
 
     def potential_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> aa.Array2D:
         """
@@ -177,7 +178,7 @@ def potential_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> aa.Array2D:
             return sum(
                 [mass.potential_2d_from(grid=grid) for mass in self.profile_list]
             )
-        return np.zeros((grid.shape[0],))
+        return jnp.zeros((grid.shape[0],))
 
     def deflections_yx_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> aa.Array2D:
         """
@@ -200,7 +201,7 @@ def deflections_yx_2d_from(self, grid: aa.type.Grid2DLike, **kwargs) -> aa.Array
             return sum(
                 [mass.deflections_yx_2d_from(grid=grid) for mass in self.profile_list]
             )
-        return np.zeros((grid.shape[0], 2))
+        return jnp.zeros((grid.shape[0], 2))
 
     def lp_instance_from(self, linear_light_profile_intensity_dict: Dict):
         light_profile_list = []