Merge pull request #501 from mirkoklukas/Type_Signature_Fix

Type signature of HeterogeneousMixture, Closes #500
probcomp · Mar 15, 2023 · 262f96f · 262f96f
2 parents cec0486 + 33c54dc
commit 262f96f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/modeling_library/mixture.jl b/src/modeling_library/mixture.jl
@@ -178,7 +178,7 @@ Gen.is_discrete(dist::HeterogeneousMixture) = dist.is_discrete
 
 const MIXTURE_WRONG_NUM_COMPONENTS_ERR = "the length of the weights vector does not match the number of mixture components"
 
-function HeterogeneousMixture(distributions::Vector{Distribution{T}}) where {T}
+function HeterogeneousMixture(distributions::Vector{D}) where {T, D <: Distribution{T}}
     _has_output_grad = true
     _has_argument_grads = Bool[true] # weights
     _is_discrete = true