Merge pull request #79 from agerlach/gd_dispatch

fix dispatch for 4 argument `GenericDistribution`
SciML · Sep 6, 2022 · 441330b · 441330b
2 parents b8865e4 + 20aa705
commit 441330b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/distribution_utils.jl b/src/distribution_utils.jl
@@ -12,7 +12,7 @@ struct GenericDistribution{TF, TRF, TLB, TUB}
     ub::TUB
 end
 
-function GenericDistribution(d, ds...)
+function GenericDistribution(d::Distributions.Sampleable, ds...)
     dists = (d, ds...)
     pdf_func(x) = exp(sum(logpdf(f, y) for (f, y) in zip(dists, x)))
     rand_func() = [rand(d) for d in dists]

diff --git a/test/interface.jl b/test/interface.jl
@@ -13,7 +13,7 @@ function (sm::SystemMap{DT})(u0, p) where {DT}
 end
 
 @testset "GenericDistribution" begin
-    dists = (Uniform(1, 2), Uniform(3, 4), Normal(0, 1))
+    dists = (Uniform(1, 2), Uniform(3, 4), Normal(0, 1), truncated(Normal(0, 1), -3, 3))
     x = [mean(d) for d in dists]
 
     pdf_f = let dists = dists