From a3b038253e04178a1481176d025e5891d4e15c7c Mon Sep 17 00:00:00 2001
From: Alfredo Braunstein <alfredo.braunstein@polito.it>
Date: Sat, 2 Mar 2019 18:16:04 +0100
Subject: [PATCH] better docs

---
 src/expectation_propagation.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/expectation_propagation.jl b/src/expectation_propagation.jl
index 0d2a258..5cc17bb 100644
--- 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