Relax Gamma Sampler types (#957)

* Relax Gamma Sampler types (#956) *relax T into Real in gamma sampler * Fixed to keep the T type dispatch, also added a test in dirichlet. * Update src/samplers/gamma.jl Co-Authored-By: Mathieu Besançon <[email protected]>
JuliaStats · Aug 22, 2019 · d493ab5 · d493ab5
1 parent 4cd28a8
commit d493ab5
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/samplers/gamma.jl b/src/samplers/gamma.jl
@@ -54,7 +54,7 @@ function GammaGDSampler(g::Gamma{T}) where {T}
         c = 0.1515/s
     end
 
-    GammaGDSampler(a,s2,s,i2s,d,q0,T(b),T(σ),T(c),scale(g))
+    GammaGDSampler(T(a), T(s2), T(s), T(i2s), T(d), T(q0), T(b), T(σ), T(c), scale(g))
 end
 
 function calc_q(s::GammaGDSampler, t)

diff --git a/test/dirichlet.jl b/test/dirichlet.jl
@@ -72,9 +72,23 @@ x = func[2](d, 10)
 @test isa(x, Matrix{Float64})
 @test size(x) == (3, 10)
 
+v = [2.0, 1.0, 3.0]
+d = Dirichlet(Float32.(v))
+
+x = func[1](d)
+@test isa(x, Vector{Float32})
+@test length(x) == 3
+
+x = func[2](d, 10)
+@test isa(x, Matrix{Float32})
+@test size(x) == (3, 10)
+
 
 # Test MLE
 
+v = [2.0, 1.0, 3.0]
+d = Dirichlet(v)
+
 n = 10000
 x = func[2](d, n)
 x = x ./ sum(x, dims=1)