Skip to content

Commit 7310731

Browse files
fix: fix SciMLStructures.replace for Discrete portion
1 parent 92bbc06 commit 7310731

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/systems/parameter_buffer.jl

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -326,7 +326,14 @@ for (Portion, field, recurse) in [(SciMLStructures.Discrete, :discrete, 2)
326326
end
327327

328328
@eval function SciMLStructures.replace(::$Portion, p::MTKParameters, newvals)
329-
@set! p.$field = split_into_buffers(newvals, p.$field, Val($recurse))
329+
@set! p.$field = $(
330+
if Portion == SciMLStructures.Discrete
331+
:(SizedVector{length(p.discrete)}(split_into_buffers(
332+
newvals, p.$field, Val($recurse))))
333+
else
334+
:(split_into_buffers(newvals, p.$field, Val($recurse)))
335+
end
336+
)
330337
if p.dependent_update_oop !== nothing
331338
raw = p.dependent_update_oop(p...)
332339
@set! p.dependent = split_into_buffers(raw, p.dependent, Val(false))

0 commit comments

Comments
 (0)