ndarray conversions for JAX arrays

Author: Jammy2211

autoarray/inversion/inversion/interferometer/mapping.py

@@ -76,7 +76,7 @@ def data_vector(self) -> np.ndarray:
         """
 
         return inversion_interferometer_util.data_vector_via_transformed_mapping_matrix_from(
-            transformed_mapping_matrix=self.operated_mapping_matrix,
+            transformed_mapping_matrix=np.array(self.operated_mapping_matrix),
             visibilities=np.array(self.data),
             noise_map=np.array(self.noise_map),
         )
@@ -152,8 +152,8 @@ def mapped_reconstructed_data_dict(
 
             visibilities = (
                 inversion_interferometer_util.mapped_reconstructed_visibilities_from(
-                    transformed_mapping_matrix=operated_mapping_matrix_list[index],
-                    reconstruction=reconstruction,
+                    transformed_mapping_matrix=np.array(operated_mapping_matrix_list[index]),
+                    reconstruction=np.array(reconstruction),
                 )
             )
 

test_autoarray/inversion/inversion/interferometer/test_inversion_interferometer_util.py

@@ -272,8 +272,8 @@ def test__identical_inversion_values_for_two_methods():
         == inversion_mapping_matrices.regularization_matrix
     ).all()
 
-    assert inversion_w_tilde.data_vector == pytest.approx(
-        inversion_mapping_matrices.data_vector, 1.0e-8
+    assert inversion_w_tilde.data_vector.array == pytest.approx(
+        inversion_mapping_matrices.data_vector.array, 1.0e-8
     )
     assert inversion_w_tilde.curvature_matrix == pytest.approx(
         inversion_mapping_matrices.curvature_matrix, 1.0e-8
@@ -384,8 +384,8 @@ def test__identical_inversion_source_and_image_loops():
     assert inversion_image_loop.reconstruction == pytest.approx(
         inversion_source_loop.reconstruction, 1.0e-2
     )
-    assert inversion_image_loop.mapped_reconstructed_image == pytest.approx(
-        inversion_source_loop.mapped_reconstructed_image, 1.0e-2
+    assert inversion_image_loop.mapped_reconstructed_image.array == pytest.approx(
+        inversion_source_loop.mapped_reconstructed_image.array, 1.0e-2
     )
     assert inversion_image_loop.mapped_reconstructed_data == pytest.approx(
         inversion_source_loop.mapped_reconstructed_data, 1.0e-2