Skip to content

Commit 6d9c333

Browse files
authored
Update model.jl
1 parent b72c70a commit 6d9c333

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/model.jl

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -397,15 +397,15 @@ Returns both the return-value of the original model, and the resulting varinfo.
397397
The method resets the log joint probability of `varinfo` and increases the evaluation
398398
number of `sampler`.
399399
"""
400-
function evaluate!!(model::Model, varinfo::AbstractVarInfo, context::AbstractContext)
400+
function AbstractPPL.evaluate!!(model::Model, varinfo::AbstractVarInfo, context::AbstractContext)
401401
return if use_threadsafe_eval(context, varinfo)
402402
evaluate_threadsafe!!(model, varinfo, context)
403403
else
404404
evaluate_threadunsafe!!(model, varinfo, context)
405405
end
406406
end
407407

408-
function evaluate!!(
408+
function AbstractPPL.evaluate!!(
409409
model::Model,
410410
rng::Random.AbstractRNG,
411411
varinfo::AbstractVarInfo=VarInfo(),
@@ -415,21 +415,21 @@ function evaluate!!(
415415
return evaluate!!(model, varinfo, SamplingContext(rng, sampler, context))
416416
end
417417

418-
evaluate!!(model::Model, context::AbstractContext) = evaluate!!(model, VarInfo(), context)
418+
AbstractPPL.evaluate!!(model::Model, context::AbstractContext) = evaluate!!(model, VarInfo(), context)
419419

420-
function evaluate!!(model::Model, args...)
420+
function AbstractPPL.evaluate!!(model::Model, args...)
421421
return evaluate!!(model, Random.GLOBAL_RNG, args...)
422422
end
423423

424424
# without VarInfo
425-
function evaluate!!(
425+
function AbstractPPL.evaluate!!(
426426
model::Model, rng::Random.AbstractRNG, sampler::AbstractSampler, args...
427427
)
428428
return evaluate!!(model, rng, VarInfo(), sampler, args...)
429429
end
430430

431431
# without VarInfo and without AbstractSampler
432-
function evaluate!!(model::Model, rng::Random.AbstractRNG, context::AbstractContext)
432+
function AbstractPPL.evaluate!!(model::Model, rng::Random.AbstractRNG, context::AbstractContext)
433433
return evaluate!!(model, rng, VarInfo(), SampleFromPrior(), context)
434434
end
435435

0 commit comments

Comments
 (0)