Skip to content

Commit 90455c8

Browse files
Update src/multivariate/dirichlet.jl
Co-authored-by: David Widmann <devmotion@users.noreply.github.com>
1 parent 1348792 commit 90455c8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/multivariate/dirichlet.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -397,7 +397,7 @@ function ChainRulesCore.rrule(DT::Union{Type{Dirichlet{T}}, Type{Dirichlet}}, al
397397
return d, dirichlet_pullback
398398
end
399399

400-
function ChainRulesCore.frule((_, Δd, Δx), ::typeof(_logpdf), d::Dirichlet, x::AbstractVector{T}) where {T}
400+
function ChainRulesCore.frule((_, Δd, Δx), ::typeof(_logpdf), d::Dirichlet, x::AbstractVector{<:Real})
401401
lp = _logpdf(d, x)
402402
if !insupport(d, x)
403403
return (lp, zero(lp))

0 commit comments

Comments
 (0)