@@ -163,37 +163,20 @@ function has_varnamedvector(vi::VarInfo)
163
163
(vi isa TypedVarInfo && any (Base. Fix2 (isa, VarNamedVector), values (vi. metadata)))
164
164
end
165
165
166
- # TODO : We should just remove this function in favour of passing `SamplingContext` directly.
167
- function untyped_varinfo (
168
- rng:: Random.AbstractRNG ,
169
- model:: Model ,
170
- sampler:: AbstractSampler = SampleFromPrior (),
171
- context:: AbstractContext = DefaultContext (),
172
- metadata:: Union{Metadata,VarNamedVector} = Metadata (),
173
- )
174
- return untyped_varinfo (model, SamplingContext (rng, sampler, context), metadata)
175
- end
176
- function untyped_varinfo (
177
- model:: Model , args:: Union{AbstractSampler,AbstractContext,Metadata,VarNamedVector} ...
178
- )
179
- return untyped_varinfo (Random. default_rng (), model, args... )
180
- end
181
-
182
166
"""
183
- untyped_varinfo([rng, ] model[, context, metadata])
167
+ untyped_varinfo(model[, context, metadata])
184
168
185
169
Return an untyped varinfo object for the given `model` and `context`.
186
170
187
171
# Arguments
188
- - `rng::Random.AbstractRNG`: The random number generator to use. Default: `Random.default_rng()`.
189
172
- `model::Model`: The model for which to create the varinfo object.
190
- - `context::AbstractContext`: The context in which to evaluate the model. Default: `DefaultContext ()`.
173
+ - `context::AbstractContext`: The context in which to evaluate the model. Default: `SamplingContext ()`.
191
174
- `metadata::Union{Metadata,VarNamedVector}`: The metadata to use for the varinfo object.
192
175
Default: `Metadata()`.
193
176
"""
194
177
function untyped_varinfo (
195
178
model:: Model ,
196
- context:: AbstractContext ,
179
+ context:: AbstractContext = SamplingContext () ,
197
180
metadata:: Union{Metadata,VarNamedVector} = Metadata (),
198
181
)
199
182
varinfo = VarInfo (metadata)
@@ -203,7 +186,7 @@ function untyped_varinfo(
203
186
end
204
187
205
188
"""
206
- typed_varinfo([rng, ] model[, context, metadata])
189
+ typed_varinfo(model[, context, metadata])
207
190
208
191
Return a typed varinfo object for the given `model`, `sampler` and `context`.
209
192
@@ -221,7 +204,7 @@ function VarInfo(
221
204
context:: AbstractContext = DefaultContext (),
222
205
metadata:: Union{Metadata,VarNamedVector} = Metadata (),
223
206
)
224
- return typed_varinfo (rng, model , sampler, context, metadata)
207
+ return typed_varinfo (model, SamplingContext (rng , sampler, context) , metadata)
225
208
end
226
209
VarInfo (model:: Model , args... ) = VarInfo (Random. default_rng (), model, args... )
227
210
0 commit comments