similar small fix to Z arm coadding in coadd_arms_forest

p-slash · Jul 9, 2024 · d757815 · d757815
1 parent a4c2aac
commit d757815
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/py/qsonic/spectrum.py b/py/qsonic/spectrum.py
@@ -611,8 +611,10 @@ def coadd_arms_forest(
         min_wave = np.min([wave[0] for wave in self.forestwave.values()])
         max_wave = np.max([wave[-1] for wave in self.forestwave.values()])
 
-        nwaves = int((max_wave - min_wave) / self.dwave + 0.1) + 1
-        coadd_wave = np.linspace(min_wave, max_wave, nwaves)
+        nwaves = round((max_wave - min_wave) / self.dwave + 0.1)
+        coadd_wave = np.linspace(
+            min_wave, min_wave + nwaves * self.dwave, nwaves + 1)
+        nwaves += 1
         coadd_flux = np.zeros(nwaves)
         coadd_ivar = np.zeros(nwaves)
         coadd_norm = np.zeros(nwaves)