Collapse 3D y before calculating initial model guess for gaussian model

spacetelescope · Nov 15, 2024 · ffd2de0 · ffd2de0
1 parent f715ffe
commit ffd2de0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/jdaviz/configs/default/plugins/model_fitting/initializers.py b/jdaviz/configs/default/plugins/model_fitting/initializers.py
@@ -188,6 +188,10 @@ def initialize(self, instance, x, y):
             The initialized model.
         """
 
+        if y.ndim == 3:
+            # For cube fitting, need to collapse before these calculations
+            y = np.nanmean(y, axis=(0, 1))
+
         # X centroid estimates the position
         centroid = np.sum(x * y) / np.sum(y)