black

Author: Jammy2211

autoarray/structures/grids/irregular_2d.py

@@ -45,8 +45,13 @@ def __init__(self, values: Union[np.ndarray, List]):
         if type(values) is list:
             if isinstance(values[0], Grid2DIrregular):
                 values = values
+            elif isinstance(values[0], jnp.ndarray):
+                values = jnp.asarray(values)
             else:
-                values = np.asarray(values)
+                try:
+                    values = np.asarray(values)
+                except ValueError:
+                    pass
 
         super().__init__(values)
 

test_autoarray/structures/grids/test_irregular_2d.py

@@ -95,7 +95,9 @@ def test__furthest_distances_to_other_coordinates():
 def test__grid_of_closest_from():
     grid = aa.Grid2DIrregular(values=[(0.0, 0.0), (0.0, 1.0)])
 
-    grid_of_closest = grid.grid_of_closest_from(grid_pair=aa.Grid2DIrregular(np.array([[0.0, 0.1]])))
+    grid_of_closest = grid.grid_of_closest_from(
+        grid_pair=aa.Grid2DIrregular(np.array([[0.0, 0.1]]))
+    )
 
     assert (grid_of_closest == np.array([[0.0, 0.0]])).all()
 
@@ -106,7 +108,9 @@ def test__grid_of_closest_from():
     assert (grid_of_closest == np.array([[0.0, 0.0], [0.0, 0.0], [0.0, 0.0]])).all()
 
     grid_of_closest = grid.grid_of_closest_from(
-        grid_pair=aa.Grid2DIrregular(np.array([[0.0, 0.1], [0.0, 0.2], [0.0, 0.9], [0.0, -0.1]]))
+        grid_pair=aa.Grid2DIrregular(
+            np.array([[0.0, 0.1], [0.0, 0.2], [0.0, 0.9], [0.0, -0.1]])
+        )
     )
 
     assert (