File tree Expand file tree Collapse file tree 3 files changed +3
-3
lines changed
autoarray/inversion/inversion Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments