From c678ffc6200d63c32562ddfdd9bc5fb764b9ef7c Mon Sep 17 00:00:00 2001 From: Markus Hauru Date: Mon, 9 Sep 2024 17:31:29 +0100 Subject: [PATCH] Fix Zygote pullback depwarning --- 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