fix

Author: rhayes777

autoarray/structures/triangles/coordinate_array/jax_coordinate_array.py

@@ -225,15 +225,14 @@ def for_indexes(self, indexes: np.ndarray) -> "CoordinateArrayTriangles":
         mask = indexes == -1
         safe_indexes = np.where(mask, 0, indexes)
         coordinates = np.take(self.coordinates, safe_indexes, axis=0)
-        # coordinates = np.where(mask[:, None], np.nan, coordinates)
+        coordinates = np.where(mask[:, None], np.nan, coordinates)
 
         return CoordinateArrayTriangles(
             coordinates=coordinates,
             side_length=self.side_length,
             y_offset=self.y_offset,
             x_offset=self.x_offset,
             flipped=self.flipped,
-            mask=mask,
         )
 
     def containing_indices(self, shape: np.ndarray) -> np.ndarray: