fix: don't normalize flux if it sums to zero

astro-group-bristol · Jun 23, 2024 · a88f10a · a88f10a
1 parent 39bf3ff
commit a88f10a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/transfer-functions/utils.jl b/src/transfer-functions/utils.jl
@@ -132,7 +132,9 @@ function _normalize!(flux::AbstractVector{T}, grid) where {T}
         flux[i] = flux[i] / ḡ
         Σflux += flux[i]
     end
-    @. flux = flux / Σflux
+    if Σflux > 0
+        @. flux = flux / Σflux
+    end
     flux
 end
 
@@ -143,6 +145,8 @@ function _normalize!(flux::AbstractMatrix{T}, grid) where {T}
         @. flux[i, :] = flux[i, :] / ḡ
         Σflux += sum(flux[i, :])
     end
-    @. flux = flux / Σflux
+    if Σflux > 0
+        @. flux = flux / Σflux
+    end
     flux
 end