diff --git a/autoarray/inversion/inversion/imaging/mapping.py b/autoarray/inversion/inversion/imaging/mapping.py index a03b0a3a..00c618eb 100644 --- a/autoarray/inversion/inversion/imaging/mapping.py +++ b/autoarray/inversion/inversion/imaging/mapping.py @@ -98,7 +98,6 @@ def curvature_matrix(self): settings=self.settings, add_to_curvature_diag=True, no_regularization_index_list=self.no_regularization_index_list, - use_mixed_precision=self.settings.use_mixed_precision, xp=self._xp, ) diff --git a/autoarray/inversion/inversion/interferometer/mapping.py b/autoarray/inversion/inversion/interferometer/mapping.py index e516bf0d..f1455b93 100644 --- a/autoarray/inversion/inversion/interferometer/mapping.py +++ b/autoarray/inversion/inversion/interferometer/mapping.py @@ -88,12 +88,14 @@ def curvature_matrix(self) -> np.ndarray: real_curvature_matrix = inversion_util.curvature_matrix_via_mapping_matrix_from( mapping_matrix=self.operated_mapping_matrix.real, noise_map=self.noise_map.real, + settings=self.settings, xp=self._xp, ) imag_curvature_matrix = inversion_util.curvature_matrix_via_mapping_matrix_from( mapping_matrix=self.operated_mapping_matrix.imag, noise_map=self.noise_map.imag, + settings=self.settings, xp=self._xp, ) diff --git a/autoarray/inversion/inversion/inversion_util.py b/autoarray/inversion/inversion/inversion_util.py index 92f65b06..7d2e6400 100644 --- a/autoarray/inversion/inversion/inversion_util.py +++ b/autoarray/inversion/inversion/inversion_util.py @@ -83,7 +83,6 @@ def curvature_matrix_via_mapping_matrix_from( add_to_curvature_diag: bool = False, no_regularization_index_list: Optional[List] = None, settings: "SettingsInversion" = SettingsInversion(), - use_mixed_precision: bool = False, xp=np, ) -> np.ndarray: """ @@ -105,7 +104,7 @@ def curvature_matrix_via_mapping_matrix_from( else: # Choose compute dtype - compute_dtype = xp.float32 if use_mixed_precision else xp.float64 + compute_dtype = xp.float32 if settings.use_mixed_precision else xp.float64 out_dtype = xp.float64 # always return float64 for downstream stability A = mapping_matrix