We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
step
1 parent e55a8e8 commit 81443d8Copy full SHA for 81443d8
src/integrator.jl
@@ -226,10 +226,8 @@ function step(
226
ϵ = fwd ? step_size(lf) : -step_size(lf)
227
ϵ = ϵ'
228
229
- res = if FullTraj
230
- Vector{P}(undef, n_steps)
231
- else
232
- Vector{P}(undef, 1)
+ if FullTraj
+ res = Vector{P}(undef, n_steps)
233
end
234
235
(; θ, r) = z
@@ -252,20 +250,18 @@ function step(
252
250
# Update result
253
251
if FullTraj
254
res[i] = z
255
256
- res[1] = z
257
258
if !isfinite(z)
259
# Remove undef
260
261
- res = res[isassigned.(Ref(res), 1:n_steps)]
+ resize!(res, i)
262
263
break
264
265
266
return if FullTraj
267
res
268
else
269
- first(res)
+ z
270
271
0 commit comments