black

Author: Jammy2211

autoarray/inversion/inversion/abstract.py

@@ -757,7 +757,10 @@ def mapper_operated_mapping_matrix_dict(self) -> Dict:
         raise NotImplementedError
 
     def max_pixel_list_from(
-        self, total_pixels: int = 1, filter_neighbors: bool = False, mapper_index : int = 0
+        self,
+        total_pixels: int = 1,
+        filter_neighbors: bool = False,
+        mapper_index: int = 0,
     ) -> List[List[int]]:
         """
         Returns a list of lists of the maximum cell or pixel values in the mapper.
@@ -814,7 +817,7 @@ def max_pixel_list_from(
 
         return max_pixel_list
 
-    def max_pixel_centre(self, mapper_index : int = 0) -> Grid2DIrregular:
+    def max_pixel_centre(self, mapper_index: int = 0) -> Grid2DIrregular:
         """
         Returns the centre of the brightest pixel in the mapper values.
 
@@ -831,4 +834,4 @@ def max_pixel_centre(self, mapper_index : int = 0) -> Grid2DIrregular:
             values=[mapper.source_plane_mesh_grid.array[max_pixel]]
         )
 
-        return max_pixel_centre
+        return max_pixel_centre

autoarray/inversion/plot/inversion_plotters.py

@@ -416,7 +416,7 @@ def subplot_mappings(
         pix_indexes = self.inversion.max_pixel_list_from(
             total_pixels=total_pixels,
             filter_neighbors=True,
-            mapper_index=pixelization_index
+            mapper_index=pixelization_index,
         )
 
         indexes = mapper.slim_indexes_for_pix_indexes(pix_indexes=pix_indexes)

test_autoarray/inversion/inversion/test_abstract.py

@@ -522,8 +522,7 @@ def test__max_pixel_list_from_and_centre():
     )
 
     inversion = aa.m.MockInversion(
-        reconstruction=np.array([2.0, 3.0, 5.0, 0.0]),
-        linear_obj_list=[mapper]
+        reconstruction=np.array([2.0, 3.0, 5.0, 0.0]), linear_obj_list=[mapper]
     )
 
     assert inversion.max_pixel_list_from(total_pixels=2)[0] == [
@@ -554,12 +553,10 @@ def test__max_pixel_list_from__filter_neighbors():
 
     inversion = aa.m.MockInversion(
         reconstruction=np.array([5.0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8]),
-        linear_obj_list=[mapper]
+        linear_obj_list=[mapper],
     )
 
-    pixel_list = inversion.max_pixel_list_from(
-        total_pixels=9, filter_neighbors=True
-    )
+    pixel_list = inversion.max_pixel_list_from(total_pixels=9, filter_neighbors=True)
 
     assert pixel_list[0] == [
         0,