Skip to content

Commit f5804ca

Browse files
committed
Update TCML_TechnionIIT_lsqlm.py
1 parent 5092b4f commit f5804ca

1 file changed

Lines changed: 11 additions & 4 deletions

File tree

src/standardized/TCML_TechnionIIT_lsqlm.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -71,15 +71,22 @@ def ivim_fit(self, signals, bvalues, **kwargs):
7171
initial_guess = [self.initial_guess["D"], self.initial_guess["Dp"], self.initial_guess["f"], self.initial_guess["S0"]]
7272
fit_results = self.fit_least_squares(bvalues, np.array(signals)[:,np.newaxis], initial_guess)
7373

74+
def get_scalar(val):
75+
"""Convert value to Python scalar, handling numpy arrays."""
76+
if isinstance(val, np.ndarray):
77+
return float(val.item())
78+
return float(val)
79+
7480
results = {}
75-
if fit_results[0].size >0:
76-
results["D"] = fit_results[0]
77-
results["f"] = fit_results[2]
78-
results["Dp"] = fit_results[1]
81+
if fit_results[0].size > 0:
82+
results["D"] = get_scalar(fit_results[0])
83+
results["f"] = get_scalar(fit_results[2])
84+
results["Dp"] = get_scalar(fit_results[1])
7985
else:
8086
results["D"] = 0
8187
results["f"] = 0
8288
results["Dp"] = 0
89+
8390
results = self.D_and_Ds_swap(results)
8491

8592
return results

0 commit comments

Comments
 (0)