Implement getindex for UntypedVarInfo

TuringLang · Oct 29, 2024 · 83d5b65 · 83d5b65
1 parent 18af48a
commit 83d5b65
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/varinfo.jl b/src/varinfo.jl
@@ -1694,6 +1694,10 @@ 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.
 """
+function getindex(vi::UntypedVarInfo, spl::Sampler)
+    indices = _getranges(vi, spl)
+    return copy(vi.metadata.vals[indices])
+end
 getindex(vi::VarInfo, spl::Sampler) = copy(getindex_internal(vi, _getranges(vi, spl)))
 function getindex(vi::TypedVarInfo, spl::Sampler)
     # Gets the ranges as a NamedTuple