Fix pluralisation in progress message

TuringLang · Nov 29, 2024 · 06696bd · 06696bd
1 parent bc760b0
commit 06696bd
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/sample.jl b/src/sample.jl
@@ -1,6 +1,8 @@
 # Default implementations of `sample`.
 const PROGRESS = Ref(true)
 
+_pluralise(n; singular="", plural="s") = n == 1 ? singular : plural
+
 """
     setprogress!(progress::Bool; silent::Bool=false)
 
@@ -372,7 +374,7 @@ function mcmcsample(
     N::Integer,
     nchains::Integer;
     progress=PROGRESS[],
-    progressname="Sampling ($(min(nchains, Threads.nthreads())) threads)",
+    progressname="Sampling ($(min(nchains, Threads.nthreads())) thread$(_pluralise(min(nchains, Threads.nthreads())))",
     initial_params=nothing,
     initial_state=nothing,
     kwargs...,
@@ -489,7 +491,7 @@ function mcmcsample(
     N::Integer,
     nchains::Integer;
     progress=PROGRESS[],
-    progressname="Sampling ($(Distributed.nworkers()) processes)",
+    progressname="Sampling ($(Distributed.nworkers()) process$(_pluralise(Distributed.nworkers(); plural="es")))",
     initial_params=nothing,
     initial_state=nothing,
     kwargs...,