black

Author: Jammy2211

autoarray/structures/triangles/abstract.py

@@ -3,7 +3,6 @@
 import numpy as np
 
 from autoarray import Grid2D
-from autoarray.structures.triangles.shape import Shape
 
 HEIGHT_FACTOR = 3**0.5 / 2
 

autoarray/structures/triangles/array.py

@@ -12,7 +12,7 @@
 
 
 @register_pytree_node_class
-class ArrayTriangles:
+class ArrayTriangles(AbstractTriangles):
     def __init__(
         self,
         indices,
@@ -118,44 +118,6 @@ def add_vertex(v):
             max_containing_size=max_containing_size,
         )
 
-    @classmethod
-    def for_grid(
-        cls,
-        grid: Grid2D,
-        **kwargs,
-    ) -> "AbstractTriangles":
-        """
-        Create a grid of equilateral triangles from a regular grid.
-
-        Parameters
-        ----------
-        grid
-            The regular grid to convert to a grid of triangles.
-
-        Returns
-        -------
-        The grid of triangles.
-        """
-
-        scale = grid.pixel_scale
-
-        y = grid[:, 0]
-        x = grid[:, 1]
-
-        y_min = y.min()
-        y_max = y.max()
-        x_min = x.min()
-        x_max = x.max()
-
-        return cls.for_limits_and_scale(
-            y_min,
-            y_max,
-            x_min,
-            x_max,
-            scale,
-            **kwargs,
-        )
-
     @property
     def indices(self):
         return self._indices

autoarray/structures/triangles/coordinate_array.py

@@ -1,8 +1,11 @@
+from abc import ABC
+
 import numpy as np
 import jax.numpy as jnp
 import jax
 
 from autoarray.structures.triangles.abstract import HEIGHT_FACTOR
+from autoarray.structures.triangles.abstract import AbstractTriangles
 from autoarray.structures.triangles.array import ArrayTriangles
 from autoarray.numpy_wrapper import register_pytree_node_class
 from autoconf import cached_property
@@ -11,7 +14,7 @@
 
 
 @register_pytree_node_class
-class CoordinateArrayTriangles:
+class CoordinateArrayTriangles(AbstractTriangles, ABC):
 
     def __init__(
         self,