Fix Unitful rearrange_to_3Darray

briochemc · Feb 15, 2020 · 2375785 · 2375785 · briochemc · Feb 15, 2020
1 parent 6628997
commit 2375785
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "OceanGrids"
 uuid = "cfe838f4-859f-11e9-2ea1-df7d4e7c3537"
 authors = ["Benoit Pasquier <pasquieb@uci.edu>"]
-version = "0.1.9"
+version = "0.1.10"
 
 [deps]
 Interpolations = "a98d9a8b-a2ab-59e6-89dd-64a1c18fca59"

diff --git a/src/OceanGrids.jl b/src/OceanGrids.jl
@@ -339,9 +339,9 @@ export latvec, lonvec, depthvec, latlondepthvecs
 Returns a 3D array of `x` rearranged to the `true` entries of `wet3D`.
 Entries where `wet3D` is `false` are filled with `NaN`s.
 """
-function rearrange_into_3Darray(x, wet3D::BitArray)
+function rearrange_into_3Darray(x::Vector{T}, wet3D::BitArray) where T
     iwet = findall(wet3D)
-    x3d = fill(NaN, size(wet3D))
+    x3d = fill(NaN * unit(T), size(wet3D))
     x3d[iwet] .= x
     return x3d
 end