Switch Print/Save order

Project.toml

@@ -1,7 +1,7 @@
 name = "Baytes"
 uuid = "72ddfcfc-6e9d-43df-829b-7aed7c549d4f"
 authors = ["Patrick Aschermayr <p.aschermayr@gmail.com>"]
-version = "0.1.8"
+version = "0.1.9"
 
 [deps]
 ArgCheck = "dce04be8-c92d-5529-be00-80e4d2c0e197"

src/sampling/sample.jl

@@ -61,18 +61,18 @@ function sample(
     ## Loop through iterations
     println("Sampling starts...")
     propose!(_rng, trace, algorithmᵛ, modelᵛ, data)
+    ## Save output
+    if safeoutput
+        println("Saving trace, initial model and algorithm.")
+        savetrace(trace, model, algorithmᵛ)
+    end
     ## Print diagnostics
-    println("Sampling finished, printing diagnostics and saving trace.")
     if printoutput
+        println("Sampling finished, printing diagnostics and saving trace.")
         ## Assign relevant parameter for printing and print summary to REPL.
         transform = TraceTransform(trace, model)
         summary(trace, algorithmᵛ, transform, printdefault)
     end
-    ## Save output
-    if safeoutput
-        println("Saving trace, initial model and algorithm.")
-        savetrace(trace, model, algorithmᵛ)
-    end
     ## Return trace and algorithm
     return trace, algorithmᵛ
 end
@@ -111,18 +111,18 @@ function sample!(iterations::Integer,
     ## Loop through iterations
     println("Sampling starts...")
     propose!(_rng, trace_new, algorithmᵛ, modelᵛ, data)
+    ## Save output
+    if safeoutput
+        println("Saving trace, initial model and algorithm.")
+        savetrace(trace_new, model, algorithmᵛ)
+    end
     ## Print diagnostics
-    println("Sampling finished, printing diagnostics.")
     if printoutput
+        println("Sampling finished, printing diagnostics.")
         ## Assign relevant parameter for printing and print summary to REPL.
         transform = TraceTransform(trace_new, model)
         summary(trace_new, algorithmᵛ, transform, printdefault)
     end
-    ## Save output
-    if safeoutput
-        println("Saving trace, initial model and algorithm.")
-        savetrace(trace_new, model, algorithmᵛ)
-    end
     ## Return trace and algorithm
     return trace_new, algorithmᵛ
 end