zebra/pyzebra
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Abort fit if there is no data in range

Browse Source
This commit is contained in:
usov_i 2021-10-05 14:24:05 +02:00
parent 960ce0a534
commit 3acd57adb9
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pyzebra/ccl_process.py
View File

@ -147,6 +147,10 @@ def fit_scan(scan, model_dict, fit_from=None, fit_to=None):
# apply fitting range # apply fitting range
fit_ind = (fit_from <= x_fit) & (x_fit <= fit_to) fit_ind = (fit_from <= x_fit) & (x_fit <= fit_to)
if not np.any(fit_ind):
print(f"No data in fit range for scan {scan['idx']}")
return
y_fit = y_fit[fit_ind] y_fit = y_fit[fit_ind]
y_err = y_err[fit_ind] y_err = y_err[fit_ind]
x_fit = x_fit[fit_ind] x_fit = x_fit[fit_ind]
@ -201,6 +205,9 @@ def fit_scan(scan, model_dict, fit_from=None, fit_to=None):
def get_area(scan, area_method, lorentz): def get_area(scan, area_method, lorentz):
if "fit" not in scan:
return
if area_method not in AREA_METHODS: if area_method not in AREA_METHODS:
raise ValueError(f"Unknown area method: {area_method}.") raise ValueError(f"Unknown area method: {area_method}.")