Incorporate suggestions from review

Author: devmotion

src/threadsafe.jl

@@ -17,7 +17,7 @@ ThreadSafeVarInfo(vi::ThreadSafeVarInfo) = vi
 # just update the array of log probabilities.
 function acclogp!(vi::ThreadSafeVarInfo, logp)
     vi.logps[Threads.threadid()] += logp
-    return getlogp(vi)
+    return vi
 end
 
 # The current log probability of the variables has to be computed from

src/varinfo.jl

@@ -602,15 +602,21 @@ getlogp(vi::AbstractVarInfo) = vi.logp[]
 Set the log of the joint probability of the observed data and parameters sampled in
 `vi` to `logp`.
 """
-setlogp!(vi::VarInfo, logp) = vi.logp[] = logp
+function setlogp!(vi::VarInfo, logp)
+    vi.logp[] = logp
+    return vi
+end
 
 """
     acclogp!(vi::VarInfo, logp)
 
 Add `logp` to the value of the log of the joint probability of the observed data and
 parameters sampled in `vi`.
 """
-acclogp!(vi::VarInfo, logp) = vi.logp[] += logp
+function acclogp!(vi::VarInfo, logp)
+    vi.logp[] += logp
+    return vi
+end
 
 """
     resetlogp!(vi::AbstractVarInfo)