added sampletype for sparsecat

JuliaPOMDP · Oct 13, 2017 · c4236f5 · c4236f5
1 parent 276847b
commit c4236f5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/distributions/sparse_cat.jl b/src/distributions/sparse_cat.jl
@@ -68,6 +68,7 @@ Base.done(d::SparseCat{V,P}, state::Integer) where {V<:AbstractArray, P<:Abstrac
 
 Base.length(d::SparseCat) = min(length(d.vals), length(d.probs))
 Base.eltype(D::Type{SparseCat{V,P}}) where {V, P} = Pair{eltype(V), eltype(P)}
+sampletype(D::Type{SparseCat{V,P}}) where {V, P} = eltype(V)
 
 function Base.mean(d::SparseCat)
     vsum = zero(eltype(d.vals))

diff --git a/test/test_sparse_cat.jl b/test/test_sparse_cat.jl
@@ -5,6 +5,8 @@ let
     @test pdf(d, :c) == 0.0
     @test pdf(d, :a) == 0.4
     @test mode(d) == :b
+    @test sampletype(d) == Symbol
+    @test sampletype(typeof(d)) == Symbol
     @inferred rand(Base.GLOBAL_RNG, d)
 
     rng = MersenneTwister(14)