fixed issue in callback for first iteration

TuringLang · Apr 25, 2024 · 50b56ab · 50b56ab
1 parent 4bf86ec
commit 50b56ab
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/abstractmcmc.jl b/src/abstractmcmc.jl
@@ -210,7 +210,8 @@ function (cb::HMCProgressCallback)(rng, model, spl, t, state, i; nadapts = 0, kw
     κ = state.κ
     tstat = t.stat
     isadapted = tstat.is_adapt
-    if isadapted
+    # The initial transition will not much information beyond the `is_adapt` field.
+    if isadapted && haskey(tstat, :numerical_error)
         cb.num_divergent_transitions_during_adaption[] += tstat.numerical_error
     else
         cb.num_divergent_transitions[] += tstat.numerical_error