From a88f10a11a4db75465130a824b7a1870f5ec90a9 Mon Sep 17 00:00:00 2001
From: fjebaker <fergusbkr@gmail.com>
Date: Mon, 24 Jun 2024 00:07:45 +0200
Subject: [PATCH] fix: don't normalize flux if it sums to zero

---
 src/transfer-functions/utils.jl | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/transfer-functions/utils.jl b/src/transfer-functions/utils.jl
index bede6b1c..ef68cf38 100644
--- 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] / ḡ
         Σ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, :] / ḡ
         Σflux += sum(flux[i, :])
     end
-    @. flux = flux / Σflux
+    if Σflux > 0
+        @. flux = flux / Σflux
+    end
     flux
 end