diff --git a/Project.toml b/Project.toml index df46559..c3723c8 100644 --- a/Project.toml +++ b/Project.toml @@ -3,7 +3,7 @@ uuid = "7a57a42e-76ec-4ea3-a279-07e840d6d9cf" keywords = ["probablistic programming"] license = "MIT" desc = "Common interfaces for probabilistic programming" -version = "0.10.0" +version = "0.10.1" [deps] AbstractMCMC = "80f14c24-f653-4e6a-9b94-39d6b0f70001" diff --git a/src/abstractprobprog.jl b/src/abstractprobprog.jl index 32e125d..d6da0d8 100644 --- a/src/abstractprobprog.jl +++ b/src/abstractprobprog.jl @@ -128,5 +128,5 @@ end Draw a sample from the predictive distribution specified by `model` with its parameters fixed to `params`. """ function StatsBase.predict(model::AbstractProbabilisticProgram, params) - return predict(Random.default_rng(), NamedTuple, model, params) + return StatsBase.predict(Random.default_rng(), model, params) end diff --git a/test/abstractprobprog.jl b/test/abstractprobprog.jl index 00230be..083bed3 100644 --- a/test/abstractprobprog.jl +++ b/test/abstractprobprog.jl @@ -13,6 +13,12 @@ function Base.rand(rng::Random.AbstractRNG, ::Type{T}, model::RandModel) where { return nothing end +struct PredModel <: AbstractProbabilisticProgram end + +function AbstractPPL.predict(rng::Random.AbstractRNG, model::PredModel, params) + return rng +end + @testset "AbstractProbabilisticProgram" begin @testset "rand defaults" begin model = RandModel(nothing, nothing) @@ -34,4 +40,9 @@ end @test model.T === NamedTuple end end + + @testset "predict defaults" begin + model = PredModel() + @test AbstractPPL.predict(model, nothing) == Random.default_rng() + end end