Skip to content

Commit 29b8118

Browse files
CopilotJammy2211
andcommitted
Remove redundant use_mixed_precision parameter, derive from settings
Co-authored-by: Jammy2211 <23455639+Jammy2211@users.noreply.github.com>
1 parent a50e5e6 commit 29b8118

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

autoarray/inversion/inversion/imaging/mapping.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,6 @@ def curvature_matrix(self):
9898
settings=self.settings,
9999
add_to_curvature_diag=True,
100100
no_regularization_index_list=self.no_regularization_index_list,
101-
use_mixed_precision=self.settings.use_mixed_precision,
102101
xp=self._xp,
103102
)
104103

autoarray/inversion/inversion/interferometer/mapping.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,12 +88,14 @@ def curvature_matrix(self) -> np.ndarray:
8888
real_curvature_matrix = inversion_util.curvature_matrix_via_mapping_matrix_from(
8989
mapping_matrix=self.operated_mapping_matrix.real,
9090
noise_map=self.noise_map.real,
91+
settings=self.settings,
9192
xp=self._xp,
9293
)
9394

9495
imag_curvature_matrix = inversion_util.curvature_matrix_via_mapping_matrix_from(
9596
mapping_matrix=self.operated_mapping_matrix.imag,
9697
noise_map=self.noise_map.imag,
98+
settings=self.settings,
9799
xp=self._xp,
98100
)
99101

autoarray/inversion/inversion/inversion_util.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,6 @@ def curvature_matrix_via_mapping_matrix_from(
8383
add_to_curvature_diag: bool = False,
8484
no_regularization_index_list: Optional[List] = None,
8585
settings: "SettingsInversion" = SettingsInversion(),
86-
use_mixed_precision: bool = False,
8786
xp=np,
8887
) -> np.ndarray:
8988
"""
@@ -105,7 +104,7 @@ def curvature_matrix_via_mapping_matrix_from(
105104
else:
106105
# Choose compute dtype
107106

108-
compute_dtype = xp.float32 if use_mixed_precision else xp.float64
107+
compute_dtype = xp.float32 if settings.use_mixed_precision else xp.float64
109108
out_dtype = xp.float64 # always return float64 for downstream stability
110109

111110
A = mapping_matrix

0 commit comments

Comments
 (0)