Skip to content

Commit 02aed1d

Browse files
authored
Fix has_argument_grads for Poisson distribution.
Discovered in #540 (reply in thread). The implementation of `has_argument_grads` mistakenly claimed that the gradient wasn't implemented for the first argument of the Poisson distribution.
1 parent 81bb2cf commit 02aed1d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/modeling_library/distributions/poisson.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,6 @@ end
2424
is_discrete(::Poisson) = true
2525

2626
has_output_grad(::Poisson) = false
27-
has_argument_grads(::Poisson) = (false,)
27+
has_argument_grads(::Poisson) = (true,)
2828

2929
export poisson

0 commit comments

Comments
 (0)