Fix Zygote pullback depwarning (#657)

TuringLang · Sep 13, 2024 · 24a7380 · 24a7380
1 parent 6ef6a2e
commit 24a7380
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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