feat: add more output and parallelization to fit accessor

Allows dictionary of `DataArray`s as parameter to fit accessor. Now, the return `Dataset` contains the data and the best fit array. Relevant tests have been added.
kmnhan · Apr 21, 2024 · 59b35f5 · 59b35f5
1 parent ab32920
commit 59b35f5
Show file tree

Hide file tree

Showing 3 changed files with 608 additions and 72 deletions.
diff --git a/src/erlab/accessors/__init__.py b/src/erlab/accessors/__init__.py
@@ -20,13 +20,18 @@
 __all__ = [
     "ModelFitDataArrayAccessor",
     "ModelFitDatasetAccessor",
+    "ParallelFitDataArrayAccessor",
     "MomentumAccessor",
     "OffsetView",
     "ImageToolAccessor",
     "PlotAccessor",
     "SelectionAccessor",
 ]
 
-from erlab.accessors.fit import ModelFitDataArrayAccessor, ModelFitDatasetAccessor
+from erlab.accessors.fit import (
+    ModelFitDataArrayAccessor,
+    ModelFitDatasetAccessor,
+    ParallelFitDataArrayAccessor,
+)
 from erlab.accessors.kspace import MomentumAccessor, OffsetView
 from erlab.accessors.utils import ImageToolAccessor, PlotAccessor, SelectionAccessor