better docs

abraunst · Mar 2, 2019 · a3b0382 · a3b0382
1 parent 77bc9ba
commit a3b0382
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/expectation_propagation.jl b/src/expectation_propagation.jl
@@ -55,13 +55,13 @@ end
 """
     expectation_propagation(FG::FactorGraph, P::AbstractMatrix{T} = Diagonal(FG.N), d::Vector{T} = zeros(FG.N);
         maxiter::Int = 2000,
-        callback = (x...)->nothing,
+        callback = (state,iter,ε)->nothing,
         damp::T = 0.9,
         epsconv::T = 1e-6,
         maxvar::T = 1e50,
         minvar::T = 1e-50,
         state::EPState{T} = EPState{T}(FG),
-        inverter::Function = inv) where {T <: Real, P <: Prior}
+        inverter::Function = inv) where {T <: Real, P <: Prior} -> (state, converged, iter, ε)
 
 
 EP for approximate inference of