Update docs/src/tutorials/prob-interface.md

Co-authored-by: David Widmann <[email protected]>
TuringLang · Feb 29, 2024 · c901f82 · c901f82
1 parent ef668a3
commit c901f82
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/src/tutorials/prob-interface.md b/docs/src/tutorials/prob-interface.md
@@ -121,8 +121,8 @@ function cross_val(
     loss = zero(logjoint(model, rand(rng, model)))
 
     # prepare the K-folds
-    fold_size = div(length(dataset), nfolds)
-    if length(dataset) % nfolds != 0
+    fold_size, remaining = divrem(length(dataset), nfolds)
+    if remaining != 0
         error("The number of folds must divide the number of data points.")
     end
     splits = Vector{Tuple{SubArray,SubArray}}(undef, nfolds)