Skip to content

Commit c253e9b

Browse files
committed
added def of untyped_varinfo which takes just model and context
1 parent dd408ee commit c253e9b

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

src/varinfo.jl

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -175,13 +175,11 @@ function untyped_varinfo(
175175
context::AbstractContext=DefaultContext(),
176176
metadata::Union{Metadata,VarNamedVector}=Metadata(),
177177
)
178-
varinfo = VarInfo(metadata)
179-
return last(evaluate!!(model, varinfo, SamplingContext(rng, sampler, context)))
178+
return untyped_varinfo(model, SamplingContext(rng, sampler, context), metadata)
180179
end
181-
function untyped_varinfo(
182-
model::Model, args::Union{AbstractSampler,AbstractContext,Metadata,VarNamedVector}...
183-
)
184-
return untyped_varinfo(Random.default_rng(), model, args...)
180+
function untyped_varinfo(model::Model, context::AbstractContext, metadata::Union{Metadata,VarNamedVector}=Metadata())
181+
varinfo = VarInfo(metadata)
182+
return last(evaluate!!(model, varinfo, hassampler(context) ? context : SamplingContext(context)))
185183
end
186184

187185
"""

0 commit comments

Comments
 (0)