fix callback_f

impICNF · Aug 4, 2023 · 14df439 · 14df439
1 parent 3534b49
commit 14df439
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/core.jl b/src/core.jl
@@ -1,11 +1,11 @@
 # SciML interface
 
-function callback_f(ps, l, icnf::AbstractFlows, prgr::Progress, itr_n::Integer)
+function callback_f(ps, l, icnf::AbstractFlows, prgr::Progress, itr_n::AbstractVector)
     ProgressMeter.next!(
         prgr;
         showvalues = [(:loss_value, l), (:iteration, itr_n), (:last_update, Dates.now())],
     )
-    itr_n[] += one(itr_n)
+    itr_n[] += one(itr_n[])
     false
 end