implement fully tested and complete

Author: Jammy2211

autogalaxy/analysis/model_util.py

@@ -178,22 +178,14 @@ def mge_point_model_from(
     # and twice the pixel scale, with a floor to avoid taking log10 of
     # very small or non-positive values.
     min_log10_sigma = -2.0  # corresponds to 0.01 arcsec
-    max_sigma = max(2.0 * pixel_scales, 10 ** min_log10_sigma)
+    max_sigma = max(2.0 * pixel_scales, 10**min_log10_sigma)
     max_log10_sigma = np.log10(max_sigma)
 
-    log10_sigma_list = np.linspace(
-        min_log10_sigma, max_log10_sigma, total_gaussians
-    )
-    centre_0 = af.UniformPrior(
-        lower_limit=centre[0] - 0.1, upper_limit=centre[0] + 0.1
-    )
-    centre_1 = af.UniformPrior(
-        lower_limit=centre[1] - 0.1, upper_limit=centre[1] + 0.1
-    )
+    log10_sigma_list = np.linspace(min_log10_sigma, max_log10_sigma, total_gaussians)
+    centre_0 = af.UniformPrior(lower_limit=centre[0] - 0.1, upper_limit=centre[0] + 0.1)
+    centre_1 = af.UniformPrior(lower_limit=centre[1] - 0.1, upper_limit=centre[1] + 0.1)
 
-    gaussian_list = af.Collection(
-        af.Model(Gaussian) for _ in range(total_gaussians)
-    )
+    gaussian_list = af.Collection(af.Model(Gaussian) for _ in range(total_gaussians))
 
     for i, gaussian in enumerate(gaussian_list):
         gaussian.centre.centre_0 = centre_0

autogalaxy/operate/deflections.py

@@ -102,7 +102,9 @@ def from_mass_obj(cls, mass_obj):
         return cls(deflections_yx_2d_from=mass_obj.deflections_yx_2d_from)
 
     @classmethod
-    def from_tracer(cls, tracer, use_multi_plane: bool = True, plane_i: int = 0, plane_j: int = -1):
+    def from_tracer(
+        cls, tracer, use_multi_plane: bool = True, plane_i: int = 0, plane_j: int = -1
+    ):
         """
         Construct from a PyAutoLens ``Tracer`` object.
 
@@ -384,9 +386,7 @@ def jacobian_from(self, grid, xp=np) -> List:
 
         return [[a11, a12], [a21, a22]]
 
-    def convergence_2d_via_hessian_from(
-        self, grid, xp=np
-    ) -> aa.ArrayIrregular:
+    def convergence_2d_via_hessian_from(self, grid, xp=np) -> aa.ArrayIrregular:
         """
         Returns the convergence of the lensing object, which is computed from the 2D deflection angle map via the
         Hessian using the expression (see equation 56 https://inspirehep.net/literature/419263):
@@ -416,9 +416,7 @@ def convergence_2d_via_hessian_from(
 
         return aa.ArrayIrregular(values=convergence)
 
-    def shear_yx_2d_via_hessian_from(
-        self, grid, xp=np
-    ) -> ShearYX2DIrregular:
+    def shear_yx_2d_via_hessian_from(self, grid, xp=np) -> ShearYX2DIrregular:
         """
         Returns the 2D (y,x) shear vectors of the lensing object, which are computed from the 2D deflection angle map
         via the Hessian using the expressions (see equation 57 https://inspirehep.net/literature/419263):
@@ -459,9 +457,7 @@ def shear_yx_2d_via_hessian_from(
 
         return ShearYX2DIrregular(values=shear_yx_2d, grid=grid)
 
-    def magnification_2d_via_hessian_from(
-        self, grid, xp=np
-    ) -> aa.ArrayIrregular:
+    def magnification_2d_via_hessian_from(self, grid, xp=np) -> aa.ArrayIrregular:
         """
         Returns the 2D magnification map of lensing object, which is computed from the 2D deflection angle map
         via the Hessian using the expressions (see equation 60 https://inspirehep.net/literature/419263):
@@ -866,4 +862,3 @@ def einstein_mass_angular_from(
             )
 
         return einstein_mass_angular_list[0]
-

autogalaxy/plot/mass_plotter.py

@@ -121,9 +121,9 @@ def figures_2d(
             from autogalaxy.operate.deflections import OperateDeflections
 
             self.mat_plot_2d.plot_array(
-                array=OperateDeflections.from_mass_obj(self.mass_obj).magnification_2d_from(
-                    grid=self.grid
-                ),
+                array=OperateDeflections.from_mass_obj(
+                    self.mass_obj
+                ).magnification_2d_from(grid=self.grid),
                 visuals_2d=self.visuals_2d_with_critical_curves,
                 auto_labels=aplt.AutoLabels(
                     title=f"Magnification{title_suffix}",