From 02aed1d4ad195146b705086e4a2c2b827697159b Mon Sep 17 00:00:00 2001 From: ztangent Date: Wed, 4 Sep 2024 18:10:46 -0400 Subject: [PATCH] Fix `has_argument_grads` for Poisson distribution. Discovered in https://github.com/probcomp/Gen.jl/discussions/540#discussioncomment-10551055. The implementation of `has_argument_grads` mistakenly claimed that the gradient wasn't implemented for the first argument of the Poisson distribution. --- src/modeling_library/distributions/poisson.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/modeling_library/distributions/poisson.jl b/src/modeling_library/distributions/poisson.jl index efb723671..199f4ab60 100644 --- 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