Implement getindex for UntypedVarInfo

TuringLang · Oct 30, 2024 · 35e3c34 · 35e3c34
1 parent 18af48a
commit 35e3c34
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/varinfo.jl b/src/varinfo.jl
@@ -1694,6 +1694,7 @@ Return the current value(s) of the random variables sampled by `spl` in `vi`.
 
 The value(s) may or may not be transformed to Euclidean space.
 """
+getindex(vi::UntypedVarInfo, spl::Sampler) = copy(getindex(vi.metadata.vals, _getranges(vi, spl)))
 getindex(vi::VarInfo, spl::Sampler) = copy(getindex_internal(vi, _getranges(vi, spl)))
 function getindex(vi::TypedVarInfo, spl::Sampler)
     # Gets the ranges as a NamedTuple