Changed FitTrace to use a linear fitter (LinearLSQFitter) when fittin…

…g a linear trace model.
astropy · Nov 26, 2024 · 2292795 · 2292795
1 parent 8b0cd21
commit 2292795
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/specreduce/tracing.py b/specreduce/tracing.py
@@ -394,9 +394,13 @@ def _fit_trace(self, img):
             y_bins = y_bins[y_finite]
 
             # use given model to bin y-values; interpolate over all wavelengths
-            fitter = (fitting.SplineSmoothingFitter()
-                      if isinstance(self.trace_model, models.Spline1D)
-                      else fitting.LMLSQFitter())
+            if isinstance(self.trace_model, models.Spline1D):
+                fitter = fitting.SplineSmoothingFitter()
+            elif self.trace_model.linear:
+                fitter = fitting.LinearLSQFitter()
+            else:
+                fitter = fitting.LMLSQFitter()
+
             self.trace_model_fit = fitter(self.trace_model, x_bins, y_bins)
 
             trace_x = np.arange(img.shape[self._disp_axis])