fix numpy use

Jammy2211 · Dec 2, 2024 · de032ae · de032ae
1 parent 7ddb7d0
commit de032ae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autoarray/structures/triangles/coordinate_array/jax_coordinate_array.py b/autoarray/structures/triangles/coordinate_array/jax_coordinate_array.py
@@ -102,7 +102,7 @@ def centres(self) -> np.ndarray:
         centres = self.scaling_factors * self.coordinates + np.array(
             [self.x_offset, self.y_offset]
         )
-        return np.where(self.mask[:, None], np.nan, centres)
+        return self.numpy.where(self.mask[:, None], np.nan, centres)
 
     @cached_property
     def flip_mask(self) -> np.ndarray: