Fix type bug in Uniform

Fixed the bug where `rand(Uniform{T})` did not preserve the type of `Uniform{T}`.
JuliaStats · May 25, 2024 · 10f1774 · 10f1774
1 parent f852803
commit 10f1774
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/univariate/continuous/uniform.jl b/src/univariate/continuous/uniform.jl
@@ -151,7 +151,7 @@ Base.:*(c::Real, d::Uniform) = Uniform(minmax(c * d.a, c * d.b)...)
 
 #### Sampling
 
-rand(rng::AbstractRNG, d::Uniform) = d.a + (d.b - d.a) * rand(rng)
+rand(rng::AbstractRNG, d::Uniform{T}) where {T} = d.a + (d.b - d.a) * rand(rng, T)
 
 _rand!(rng::AbstractRNG, d::Uniform, A::AbstractArray{<:Real}) =
     A .= Base.Fix1(quantile, d).(rand!(rng, A))