add comment for entropy on LocationScale

TuringLang · Jun 13, 2024 · c93b5d7 · c93b5d7 · github-actions · Jun 13, 2024
1 parent 9062431
commit c93b5d7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/families/location_scale.jl b/src/families/location_scale.jl
@@ -67,6 +67,7 @@ Base.eltype(::Type{<:MvLocationScale{S, D, L}}) where {S, D, L} = eltype(D)
 function StatsBase.entropy(q::MvLocationScale)
     @unpack  location, scale, dist = q
     n_dims = length(location)
+    # `convert` is necessary because `entropy` is not type stable upstream
     n_dims*convert(eltype(location), entropy(dist)) + logdet(scale)
 end