Whoops, this needs to be a broadcast

natgeo-wong · Nov 10, 2024 · 4b87463 · 4b87463
1 parent 8c99eb3
commit 4b87463
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/extract.jl b/src/extract.jl
@@ -182,9 +182,10 @@ function extractwwtg(
 
         iwwtg = @views wwtg[:,it,ids]
         if !isnan(sum(iwwtg))
-            itrop = findlast(iwwtg!=0) + 1
+            itrop = findlast(.!iszero.(iwwtg)) + 1
             ztrop[it,ids] = z[itrop,ids]
             ptrop[it,ids] = p[itrop,ids]
+            iwwtg = @views iwwtg[1:itrop]
             iz = @views z[1:itrop,ids]
             for imode = 1 : nmode
                 iiwwtg = iwwtg .* sin.(iz./ztrop[it,ids]*pi*imode)