🎨 Add comment

arafune · Apr 27, 2024 · 5fa0519 · 5fa0519
1 parent 4b10bf5
commit 5fa0519
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/arpes/analysis/band_analysis.py b/src/arpes/analysis/band_analysis.py
@@ -215,11 +215,14 @@ def _identified_band_results_etc(
 
     Args:
         band_results (xr.DataArray): band results.
+            the value must be the array of lmfit.model.ModelResul
+            return of broadcast_model().results
         weights (tuple[float, float, float]): weight values for sigma, amplitude, center
 
     Returns:
         [TODO:description]
     """
+    band_results = band_results if isinstance(band_results, xr.DataArray) else band_results.results
     prefixes = [component.prefix for component in band_results.values[0].model.components]
 
     identified_band_results = copy.deepcopy(band_results)