Merge branch 'master' into compathelper/new_version/2024-02-29-00-08-…

…46-910-02046243722
TuringLang · Mar 1, 2024 · a6bcd00 · a6bcd00
2 parents a1276f5 + 2bc717c
commit a6bcd00
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/sampler.jl b/src/sampler.jl
@@ -158,11 +158,14 @@ function initialize_parameters!!(
         vi = invlink!!(vi, spl, model)
     end
     theta = vi[spl]
-    length(theta) == length(init_theta) ||
-        error("Provided initial value doesn't match the dimension of the model")
+    length(theta) == length(init_theta) || throw(
+        DimensionMismatch(
+            "Provided initial value size ($(length(init_theta))) doesn't match the model size ($(length(theta)))",
+        ),
+    )
 
     # Update values that are provided.
-    for i in 1:length(init_theta)
+    for i in eachindex(init_theta)
         x = init_theta[i]
         if x !== missing
             theta[i] = x