From 24a73809b1a9f0f5b4e5f907f737b57c6eaf801a Mon Sep 17 00:00:00 2001 From: Markus Hauru Date: Fri, 13 Sep 2024 23:33:53 +0200 Subject: [PATCH] Fix Zygote pullback depwarning (#657) --- ext/DynamicPPLZygoteRulesExt.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ext/DynamicPPLZygoteRulesExt.jl b/ext/DynamicPPLZygoteRulesExt.jl index 93b87e2a0..78831fdc4 100644 --- a/ext/DynamicPPLZygoteRulesExt.jl +++ b/ext/DynamicPPLZygoteRulesExt.jl @@ -19,7 +19,7 @@ ZygoteRules.@adjoint function DynamicPPL.dot_observe( DynamicPPL.increment_num_produce!(vi) return sum(map(Distributions.loglikelihood, dists, value)), vi end - return ZygoteRules.pullback(__context__, dot_observe_fallback, spl, dists, value, vi) + return ZygoteRules.pullback(dot_observe_fallback, __context__, spl, dists, value, vi) end end # module