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

Improve naming consistency with lmfit

Browse Source
This commit is contained in:
usov_i
2021-03-01 15:18:55 +01:00
parent 7c1e2fdf0c
commit 3e174f22e5
3 changed files with 35 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pyzebra/ccl_process.py
View File

@@ -96,13 +96,13 @@ def fit_scan(scan, model_dict):
model_name, _ = model_name.split("-")
prefix = f"f{model_index}_"
if model_name == "background":
_model = LinearModel(prefix=prefix, name="background")
elif model_name == "gauss":
_model = GaussianModel(prefix=prefix, name="gauss")
if model_name == "linear":
_model = LinearModel(prefix=prefix)
elif model_name == "gaussian":
_model = GaussianModel(prefix=prefix)
elif model_name == "voigt":
_model = VoigtModel(prefix=prefix)
elif model_name == "pseudovoigt":
elif model_name == "pvoigt":
_model = PseudoVoigtModel(prefix=prefix)
else:
raise ValueError(f"Unknown model name: '{model_name}'")