stuff urgh

Author: Jammy2211

.github/workflows/main.yml

@@ -66,7 +66,6 @@ jobs:
     - name: Run tests
       run: |
         export ROOT_DIR=`pwd`
-        export JAX_ENABLE_X64=True
         export PYTHONPATH=$PYTHONPATH:$ROOT_DIR/PyAutoConf
         export PYTHONPATH=$PYTHONPATH:$ROOT_DIR/PyAutoArray
         pushd PyAutoArray

autoarray/inversion/pixelization/mappers/abstract.py

@@ -370,9 +370,15 @@ def sparse_triplets_curvature(self):
             Mapping weight for each entry.
         """
 
-        rows, cols, vals = self.sparse_triplets_data
-
-        rows = self.mapper_grids.mask.fft_index_for_masked_pixel[rows]
+        rows, cols, vals = mapper_util.sparse_triplets_from(
+            pix_indexes_for_sub=self.pix_indexes_for_sub_slim_index,
+            pix_weights_for_sub=self.pix_weights_for_sub_slim_index,
+            slim_index_for_sub=self.slim_index_for_sub_slim_index,
+            fft_index_for_masked_pixel=self.mapper_grids.mask.fft_index_for_masked_pixel,
+            sub_fraction_slim=self.over_sampler.sub_fraction.array,
+            xp=self._xp,
+            return_rows_slim=False
+        )
 
         return rows, cols, vals
 

test_autoarray/inversion/inversion/imaging/test_inversion_imaging_util.py

@@ -204,8 +204,6 @@ def test__data_vector_via_weighted_data_two_methods_agree():
 
     for sub_size in range(1, 3):
 
-        print(sub_size)
-
         grid = aa.Grid2D.from_mask(mask=mask, over_sample_size=sub_size)
 
         mapper_grids = pixelization.mapper_grids_from(