grid 2d follows rules

Jammy2211 · Jammy2211 · commit b5601e82d0c1 · 2025-04-06T16:24:41.000+01:00
diff --git a/autoarray/structures/grids/uniform_2d.py b/autoarray/structures/grids/uniform_2d.py
@@ -862,7 +862,7 @@ def distances_to_coordinate_from(
         squared_distance = self.squared_distances_to_coordinate_from(
             coordinate=coordinate
         )
-        distances = np.sqrt(squared_distance.array)
+        distances = jnp.sqrt(squared_distance.array)
         return Array2D(values=distances, mask=self.mask)
 
     def grid_2d_radial_projected_shape_slim_from(

Original file line number	Diff line number	Diff line change
`@@ -862,7 +862,7 @@ def distances_to_coordinate_from(`
`862`	`862`	`squared_distance = self.squared_distances_to_coordinate_from(`
`863`	`863`	`coordinate=coordinate`
`864`	`864`	`)`
`865`		`- distances = np.sqrt(squared_distance.array)`
	`865`	`+ distances = jnp.sqrt(squared_distance.array)`
`866`	`866`	`return Array2D(values=distances, mask=self.mask)`
`867`	`867`
`868`	`868`	`def grid_2d_radial_projected_shape_slim_from(`