Make assert_timedim_last more generalized

Tokazama · Jul 6, 2020 · 33d8660 · 33d8660
1 parent a46bd3c
commit 33d8660
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 = "TimeAxes"
 uuid = "9a9fc9a6-283d-47e9-a2f6-b3a44e559ea3"
 authors = ["Zachary P. Christensen <[email protected]>"]
-version = "0.3"
+version = "0.3.1"
 
 [deps]
 AbstractFFTs = "621f4979-c628-5d54-868e-fcf4e3e8185c"

diff --git a/src/timedim.jl b/src/timedim.jl
@@ -56,9 +56,9 @@ sampling_rate(x) = 1 / time_step(x)
 
 Throw an error if the `x` has a time dimension that is not the last dimension.
 """
-@inline function assert_timedim_last(x::AbstractArray{T,N}) where {T,N}
+@inline function assert_timedim_last(x::T) where {T}
     if has_timedim(x)
-        if timedim(x) === N
+        if timedim(x) === ndims(T)
             return nothing
         else
             error("time dimension is not last")