Add exclamation marks to function names

Author: penelopeysm

src/logging.jl

@@ -18,13 +18,13 @@ struct CreateNewProgressBar{S<:AbstractString} <: AbstractProgressKwarg
         return new{typeof(name)}(name, UUIDs.uuid4())
     end
 end
-function init_progress(p::CreateNewProgressBar)
+function init_progress!(p::CreateNewProgressBar)
     ProgressLogging.@logprogress p.name nothing _id = p.uuid
 end
-function update_progress(p::CreateNewProgressBar, progress_frac)
+function update_progress!(p::CreateNewProgressBar, progress_frac)
     ProgressLogging.@logprogress p.name progress_frac _id = p.uuid
 end
-function finish_progress(p::CreateNewProgressBar)
+function finish_progress!(p::CreateNewProgressBar)
     ProgressLogging.@logprogress p.name "done" _id = p.uuid
 end
 
@@ -34,9 +34,9 @@ end
 Do not log progress at all.
 """
 struct NoLogging <: AbstractProgressKwarg end
-init_progress(::NoLogging) = nothing
-update_progress(::NoLogging, ::Any) = nothing
-finish_progress(::NoLogging) = nothing
+init_progress!(::NoLogging) = nothing
+update_progress!(::NoLogging, ::Any) = nothing
+finish_progress!(::NoLogging) = nothing
 
 """
     ExistingProgressBar
@@ -51,7 +51,7 @@ struct ExistingProgressBar{S<:AbstractString} <: AbstractProgressKwarg
     name::S
     uuid::UUIDs.UUID
 end
-function init_progress(p::ExistingProgressBar)
+function init_progress!(p::ExistingProgressBar)
     # Hacky code to reset the start timer if called from a multi-chain sampling
     # process. We need this because the progress bar is constructed in the
     # multi-chain method, i.e. if we don't do this the progress bar shows the
@@ -65,10 +65,10 @@ function init_progress(p::ExistingProgressBar)
     end
     ProgressLogging.@logprogress p.name nothing _id = p.uuid
 end
-function update_progress(p::ExistingProgressBar, progress_frac)
+function update_progress!(p::ExistingProgressBar, progress_frac)
     ProgressLogging.@logprogress p.name progress_frac _id = p.uuid
 end
-function finish_progress(p::ExistingProgressBar)
+function finish_progress!(p::ExistingProgressBar)
     ProgressLogging.@logprogress p.name "done" _id = p.uuid
 end
 
@@ -87,11 +87,11 @@ struct ChannelProgress{T<:Union{Channel{Bool},Distributed.RemoteChannel{Channel{
     channel::T
     n_updates::Int
 end
-init_progress(::ChannelProgress) = nothing
-update_progress(p::ChannelProgress, ::Any) = put!(p.channel, true)
+init_progress!(::ChannelProgress) = nothing
+update_progress!(p::ChannelProgress, ::Any) = put!(p.channel, true)
 # Note: We don't want to `put!(p.channel, false)`, because that would stop the
 # channel from being used for further updates e.g. from other chains.
-finish_progress(::ChannelProgress) = nothing
+finish_progress!(::ChannelProgress) = nothing
 
 # Add a custom progress logger if the current logger does not seem to be able to handle
 # progress logs.

src/sample.jl

@@ -169,7 +169,7 @@ function mcmcsample(
     local state
 
     @maybewithricherlogger begin
-        init_progress(progress)
+        init_progress!(progress)
         # Determine threshold values for progress logging (by default, one
         # update per 0.5% of progress, unless this has been passed in
         # explicitly)
@@ -195,7 +195,7 @@ function mcmcsample(
         # Start the progress bar.
         itotal = 1
         if itotal >= next_update
-            update_progress(progress, itotal / Ntotal)
+            update_progress!(progress, itotal / Ntotal)
             next_update += threshold
         end
 
@@ -211,7 +211,7 @@ function mcmcsample(
             # Update the progress bar.
             itotal += 1
             if itotal >= next_update
-                update_progress(progress, itotal / Ntotal)
+                update_progress!(progress, itotal / Ntotal)
                 next_update += threshold
             end
         end
@@ -237,7 +237,7 @@ function mcmcsample(
                 # Update progress bar.
                 itotal += 1
                 if itotal >= next_update
-                    update_progress(progress, itotal / Ntotal)
+                    update_progress!(progress, itotal / Ntotal)
                     next_update += threshold
                 end
             end
@@ -259,11 +259,11 @@ function mcmcsample(
             # Update the progress bar.
             itotal += 1
             if itotal >= next_update
-                update_progress(progress, itotal / Ntotal)
+                update_progress!(progress, itotal / Ntotal)
                 next_update += threshold
             end
         end
-        finish_progress(progress)
+        finish_progress!(progress)
     end
 
     # Get the sample stop time.
@@ -322,7 +322,7 @@ function mcmcsample(
     local state
 
     @maybewithricherlogger begin
-        init_progress(progress)
+        init_progress!(progress)
         # Obtain the initial sample and state.
         sample, state = if num_warmup > 0
             if initial_state === nothing
@@ -385,7 +385,7 @@ function mcmcsample(
             # Increment iteration counter.
             i += 1
         end
-        finish_progress(progress)
+        finish_progress!(progress)
     end
 
     # Get the sample stop time.
@@ -474,7 +474,7 @@ function mcmcsample(
             # by a channel, but it is not itself a ChannelProgress (because
             # ChannelProgress doesn't come with a progress bar).
             overall_progress_bar = CreateNewProgressBar(progressname)
-            init_progress(overall_progress_bar)
+            init_progress!(overall_progress_bar)
             # These are the per-chain progress bars. We generate `nchains`
             # independent UUIDs for each progress bar
             child_progresses = [
@@ -484,7 +484,7 @@ function mcmcsample(
             # ProgressLogging prints from the bottom up, and we want chain 1 to
             # show up at the top)
             for child_progress in reverse(child_progresses)
-                init_progress(child_progress)
+                init_progress!(child_progress)
             end
             updates_per_chain = nothing
         elseif progress == :overall
@@ -518,11 +518,11 @@ function mcmcsample(
                     while take!(progress_channel)
                         itotal += 1
                         if itotal >= next_update
-                            update_progress(overall_progress_bar, itotal / Ntotal)
+                            update_progress!(overall_progress_bar, itotal / Ntotal)
                             next_update += threshold
                         end
                     end
-                    finish_progress(overall_progress_bar)
+                    finish_progress!(overall_progress_bar)
                 end
             end
 
@@ -578,7 +578,7 @@ function mcmcsample(
                                 # Tell the 'main' progress bar that this chain is done.
                                 put!(progress_channel, true)
                                 # Conclude the per-chain progress bar.
-                                finish_progress(child_progresses[chainidx])
+                                finish_progress!(child_progresses[chainidx])
                             end
                             # Note that if progress == :overall, we don't need to do anything
                             # because progress on that bar is triggered by
@@ -593,7 +593,7 @@ function mcmcsample(
                         put!(progress_channel, false)
                         # Additionally stop the per-chain progress bars
                         for child_progress in child_progresses
-                            finish_progress(child_progress)
+                            finish_progress!(child_progress)
                         end
                     elseif progress == :overall
                         # Stop updating the main progress bar (either if sampling
@@ -670,7 +670,7 @@ function mcmcsample(
             chan = Channel{Bool}(Distributed.nworkers())
             progress_channel = Distributed.RemoteChannel(() -> chan)
             overall_progress_bar = CreateNewProgressBar(progressname)
-            init_progress(overall_progress_bar)
+            init_progress!(overall_progress_bar)
             # See MCMCThreads method for the rationale behind updates_per_chain.
             updates_per_chain = max(1, 400 ÷ nchains)
             child_progresses = [
@@ -694,11 +694,11 @@ function mcmcsample(
                     while take!(progress_channel)
                         itotal += 1
                         if itotal >= next_update
-                            update_progress(overall_progress_bar, itotal / Ntotal)
+                            update_progress!(overall_progress_bar, itotal / Ntotal)
                             next_update += threshold
                         end
                     end
-                    finish_progress(overall_progress_bar)
+                    finish_progress!(overall_progress_bar)
                 end
             end