black

Author: Jammy2211

autoarray/fixtures.py

@@ -67,6 +67,7 @@ def make_array_1d_7():
 def make_array_2d_7x7():
     return aa.Array2D.ones(shape_native=(7, 7), pixel_scales=(1.0, 1.0))
 
+
 def make_array_2d_rgb_7x7():
     return aa.Array2DRGB(values=np.ones((7, 7, 3)), mask=make_mask_2d_7x7())
 

autoarray/mask/derive/zoom_2d.py

@@ -296,17 +296,19 @@ def array_2d_rgb_from(self, array: Array2DRGB, buffer: int = 1) -> Array2DRGB:
         for i in range(3):
 
             extracted_array_2d = array_2d_util.extracted_array_2d_from(
-                array_2d=np.array(array.native[:,:,i]),
+                array_2d=np.array(array.native[:, :, i]),
                 y0=self.region[0] - buffer,
                 y1=self.region[1] + buffer,
                 x0=self.region[2] - buffer,
                 x1=self.region[3] + buffer,
             )
 
             if i == 0:
-                array_2d_rgb = np.zeros((extracted_array_2d.shape[0], extracted_array_2d.shape[1], 3))
+                array_2d_rgb = np.zeros(
+                    (extracted_array_2d.shape[0], extracted_array_2d.shape[1], 3)
+                )
 
-            array_2d_rgb[:,:,i] = extracted_array_2d
+            array_2d_rgb[:, :, i] = extracted_array_2d
 
         extracted_mask_2d = array_2d_util.extracted_array_2d_from(
             array_2d=np.array(self.mask),
@@ -322,4 +324,4 @@ def array_2d_rgb_from(self, array: Array2DRGB, buffer: int = 1) -> Array2DRGB:
             origin=array.mask.mask_centre,
         )
 
-        return Array2DRGB(values=array_2d_rgb.astype("int"), mask=mask)
+        return Array2DRGB(values=array_2d_rgb.astype("int"), mask=mask)

autoarray/structures/arrays/rgb.py

@@ -1,6 +1,7 @@
 from autoarray.abstract_ndarray import AbstractNDArray
 from autoarray.structures.arrays.uniform_2d import Array2D
 
+
 class Array2DRGB(Array2D):
 
     def __init__(self, values, mask):
@@ -41,4 +42,4 @@ def native(self) -> "Array2D":
         """
         Returns the RGB ndarray of shape [total_y_pixels, total_x_pixels, 3] in its `native` representation.
         """
-        return self
+        return self

test_autoarray/structures/plot/test_structure_plotters.py

@@ -183,4 +183,4 @@ def test__array_rgb(
 
     array_plotter.figure_2d()
 
-    assert path.join(plot_path, "array_rgb.png") in plot_patch.paths
+    assert path.join(plot_path, "array_rgb.png") in plot_patch.paths