Skip to content

Commit 66ec113

Browse files
fix: handle empty reduce in rrule
1 parent 6686524 commit 66ec113

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

ext/MTKChainRulesCoreExt.jl

+4-2
Original file line numberDiff line numberDiff line change
@@ -79,9 +79,11 @@ function ChainRulesCore.rrule(
7979
end
8080
newbuf = MTK.remake_buffer(indp, oldbuf, idxs, vals)
8181
tunable_idxs = reduce(
82-
vcat, (idx.idx for idx in idxs if idx.portion isa MTK.SciMLStructures.Tunable))
82+
vcat, (idx.idx for idx in idxs if idx.portion isa MTK.SciMLStructures.Tunable);
83+
init = Union{Int, AbstractVector{Int}}[])
8384
initials_idxs = reduce(
84-
vcat, (idx.idx for idx in idxs if idx.portion isa MTK.SciMLStructures.Initials))
85+
vcat, (idx.idx for idx in idxs if idx.portion isa MTK.SciMLStructures.Initials);
86+
init = Union{Int, AbstractVector{Int}}[])
8587
disc_idxs = subset_idxs(idxs, MTK.SciMLStructures.Discrete(), oldbuf.discrete)
8688
const_idxs = subset_idxs(idxs, MTK.SciMLStructures.Constants(), oldbuf.constant)
8789
nn_idxs = subset_idxs(idxs, MTK.NONNUMERIC_PORTION, oldbuf.nonnumeric)

0 commit comments

Comments
 (0)