Skip to content

Commit

Permalink
Fix has_argument_grads for Poisson distribution.
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
ztangent authored Sep 4, 2024
1 parent 81bb2cf commit 02aed1d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/modeling_library/distributions/poisson.jl
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit 02aed1d

Please sign in to comment.