black

Author: Jammy2211

autogalaxy/abstract_fit.py

@@ -15,12 +15,7 @@
 
 
 class AbstractFitInversion:
-    def __init__(
-        self,
-        model_obj,
-        settings_inversion: aa.SettingsInversion,
-        xp=np
-    ):
+    def __init__(self, model_obj, settings_inversion: aa.SettingsInversion, xp=np):
         """
         An abstract fit object which fits to datasets (e.g. imaging, interferometer) inherit from.
 
@@ -131,7 +126,9 @@ def linear_light_profile_intensity_dict(
 
             for i, light_profile in enumerate(linear_obj_func.light_profile_list):
                 if self.use_jax:
-                    linear_light_profile_intensity_dict[light_profile] = reconstruction[i]
+                    linear_light_profile_intensity_dict[light_profile] = reconstruction[
+                        i
+                    ]
                 else:
                     linear_light_profile_intensity_dict[light_profile] = float(
                         reconstruction[i]

autogalaxy/galaxy/galaxies.py

@@ -119,7 +119,10 @@ def image_2d_from(
         )
 
     def galaxy_image_2d_dict_from(
-        self, grid: aa.type.Grid2DLike, xp=np, operated_only: Optional[bool] = None,
+        self,
+        grid: aa.type.Grid2DLike,
+        xp=np,
+        operated_only: Optional[bool] = None,
     ) -> {Galaxy: np.ndarray}:
         """
         Returns a dictionary associating every `Galaxy` object with its corresponding 2D image, using the instance

autogalaxy/imaging/fit_imaging.py

@@ -78,7 +78,7 @@ def __init__(
             self=self,
             model_obj=self.galaxies,
             settings_inversion=settings_inversion,
-            xp=xp
+            xp=xp,
         )
 
         self.adapt_images = adapt_images
@@ -184,8 +184,7 @@ def galaxy_image_dict(self) -> Dict[Galaxy, np.ndarray]:
         """
 
         galaxy_image_2d_dict = self.galaxies.galaxy_image_2d_dict_from(
-            grid=self.grids.lp,
-            xp=self._xp
+            grid=self.grids.lp, xp=self._xp
         )
 
         galaxy_linear_obj_image_dict = self.galaxy_linear_obj_data_dict_from(
@@ -213,7 +212,7 @@ def galaxy_model_image_dict(self) -> Dict[Galaxy, np.ndarray]:
             grid=self.grids.lp,
             psf=self.dataset.psf,
             blurring_grid=self.grids.blurring,
-            xp=self._xp
+            xp=self._xp,
         )
 
         galaxy_linear_obj_image_dict = self.galaxy_linear_obj_data_dict_from(

autogalaxy/interferometer/fit_interferometer.py

@@ -72,7 +72,7 @@ def __init__(
             self=self,
             model_obj=self.galaxies,
             settings_inversion=settings_inversion,
-            xp=xp
+            xp=xp,
         )
 
         self.adapt_images = adapt_images
@@ -161,7 +161,9 @@ def galaxy_image_dict(self) -> Dict[Galaxy, np.ndarray]:
         For modeling, this dictionary is used to set up the `adapt_images` that adapt certain pixelizations to the
         data being fitted.
         """
-        galaxy_image_dict = self.galaxies.galaxy_image_2d_dict_from(grid=self.grids.lp, xp=self._xp)
+        galaxy_image_dict = self.galaxies.galaxy_image_2d_dict_from(
+            grid=self.grids.lp, xp=self._xp
+        )
 
         galaxy_linear_obj_image_dict = self.galaxy_linear_obj_data_dict_from(
             use_operated=False

autogalaxy/operate/image.py

@@ -414,7 +414,7 @@ def galaxy_blurred_image_2d_dict_from(
             blurred_image_2d = psf.convolved_image_from(
                 image=image_2d_not_operated,
                 blurring_image=blurring_image_2d_not_operated,
-                xp=xp
+                xp=xp,
             )
 
             image_2d_operated = galaxy_image_2d_operated_dict[galaxy_key]