Update src/metric.jl

TuringLang · Jul 8, 2024 · 920b409 · 920b409
1 parent ad2ad2c
commit 920b409
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/metric.jl b/src/metric.jl
@@ -147,6 +147,6 @@ Base.rand(
     rng::AbstractVector{<:AbstractRNG},
     metric::AbstractMetric,
     kinetic::AbstractKinetic,
-    θ,
+    θ::AbstractVecOrMat,
 ) = rand(rng, metric, kinetic)
 Base.rand(metric::AbstractMetric, kinetic::AbstractKinetic, θ::AbstractVecOrMat) = rand(metric, kinetic)