Merge pull request #541 from probcomp/fix-poisson-argument-grad

Fix `has_argument_grads` for Poisson distribution.
probcomp · Sep 5, 2024 · 5f8520b · 5f8520b
2 parents 81bb2cf + 02aed1d
commit 5f8520b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/modeling_library/distributions/poisson.jl b/src/modeling_library/distributions/poisson.jl
@@ -24,6 +24,6 @@ end
 is_discrete(::Poisson) = true
 
 has_output_grad(::Poisson) = false
-has_argument_grads(::Poisson) = (false,)
+has_argument_grads(::Poisson) = (true,)
 
 export poisson