Skip to content

Commit 24a7380

Browse files
authored
Fix Zygote pullback depwarning (#657)
1 parent 6ef6a2e commit 24a7380

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ext/DynamicPPLZygoteRulesExt.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ ZygoteRules.@adjoint function DynamicPPL.dot_observe(
1919
DynamicPPL.increment_num_produce!(vi)
2020
return sum(map(Distributions.loglikelihood, dists, value)), vi
2121
end
22-
return ZygoteRules.pullback(__context__, dot_observe_fallback, spl, dists, value, vi)
22+
return ZygoteRules.pullback(dot_observe_fallback, __context__, spl, dists, value, vi)
2323
end
2424

2525
end # module

0 commit comments

Comments
 (0)