Fix initializing linear component for cube fit

spacetelescope · Sep 16, 2024 · bd1f425 · bd1f425
1 parent 325498b
commit bd1f425
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/jdaviz/configs/default/plugins/model_fitting/initializers.py b/jdaviz/configs/default/plugins/model_fitting/initializers.py
@@ -77,6 +77,9 @@ def initialize(self, instance, x, y):
         instance : `~astropy.modeling.Model`
             The initialized model.
         """
+        if y.ndim == 3:
+            # For cube fitting, need to collapse before this calculation
+            y = np.nanmean(y, axis=(0,1))
         slope, intercept = np.polynomial.Polynomial.fit(x.value.flatten(), y.value.flatten(), 1)
 
         instance.slope.value = slope