more SampleFromUniform fixes

Author: mohdibntarek

src/utils.jl

@@ -118,7 +118,7 @@ end
 # ROBUST INITIALISATIONS
 # Uniform rand with range 2; ref: https://mc-stan.org/docs/2_19/reference-manual/initialization.html
 randrealuni() = Real(4rand()-2)
-randrealuni(args...) = map(Real, 4rand(args...)-2)
+randrealuni(args...) = map(Real, 4 .* rand(args...) .- 2)
 
 const Transformable = Union{TransformDistribution, SimplexDistribution, PDMatDistribution}
 

src/varinfo.jl

@@ -820,6 +820,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::AbstractVarInfo, spl::SampleFromPrior) = copy(getall(vi))
+getindex(vi::AbstractVarInfo, spl::SampleFromUniform) = copy(getall(vi))
 getindex(vi::UntypedVarInfo, spl::Sampler) = copy(getval(vi, _getranges(vi, spl)))
 function getindex(vi::TypedVarInfo, spl::Sampler)
     # Gets the ranges as a NamedTuple