fix saveidxs

SciML · Aug 5, 2018 · 36bacb4 · 36bacb4
1 parent 6742e25
commit 36bacb4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/dense/generic_dense.jl b/src/dense/generic_dense.jl
@@ -332,7 +332,8 @@ function ode_interpolant(Θ,dt,y₀,y₁,k,cache,idxs,T::Type{Val{TI}}) where TI
 end
 
 function ode_interpolant(Θ,dt,y₀,y₁,k,cache::OrdinaryDiffEqMutableCache,idxs,T::Type{Val{TI}}) where TI
-  if typeof(idxs) <: Number
+  if typeof(idxs) <: Number || typeof(y₀) <: Union{Number,SArray}
+    # typeof(y₀) can be these if saveidxs gives a single value
     _ode_interpolant(Θ,dt,y₀,y₁,k,cache,idxs,T)
   else
     # determine output type