Properly convert to bin edges in FormalIntegrator

tardis-sn · Jun 8, 2017 · 678186a · 678186a
1 parent 4cd961b
commit 678186a
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tardis/montecarlo/formal_integral.py b/tardis/montecarlo/formal_integral.py
@@ -27,9 +27,12 @@ def calculate_spectrum(self, frequency, N=1000):
                 ) * (frequency[1] - frequency[0])
         # Ugly hack to convert to 'bin edges'
         frequency = u.Quantity(
-                np.concatenate(
-                    [frequency.value, [0]]),
-                frequency.unit)
+                np.concatenate([
+                    frequency.value,
+                    [
+                        frequency.value[-1] + np.diff(frequency.value)[-1]
+                        ]]),
+                    frequency.unit)
         return TARDISSpectrum(
                 frequency,
                 luminosity