Merge pull request #20 from Rodot-/fast_formal_integral_subbranch

added back the z_end
tardis-sn · Jul 22, 2024 · 2c44873 · 2c44873
2 parents 9aa9983 + 90a1072
commit 2c44873
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