added back the z_end

tardis-sn · Jul 22, 2024 · 90a1072 · 90a1072
1 parent 098db98
commit 90a1072
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tardis/transport/montecarlo/formal_integral_cuda.py b/tardis/transport/montecarlo/formal_integral_cuda.py
@@ -175,6 +175,7 @@ def cuda_formal_integral(
         )  # +1 is the offset as the original is from z[1:]
 
         nu_end_idx = line_search_cuda(line_list_nu, nu_end, len(line_list_nu))
+        zend = time_explosion / C_INV * (1.0 - line_list_nu[pline] / nu)
         escat_contrib += (
             (zend - zstart)
             * escat_op