Update src/caches/abstractbeliefpropagationcache.jl

Co-authored-by: Matt Fishman <[email protected]>

Author: JoeyT1994

src/caches/abstractbeliefpropagationcache.jl

@@ -144,7 +144,7 @@ function map_factors(f, bpc::AbstractBeliefPropagationCache, vs=vertices(bpc))
   return bpc
 end
 function adapt_messages(to, bpc::AbstractBeliefPropagationCache, args...)
-  map_messages(adapt(to), bpc, args...)
+  return map_messages(adapt(to), bpc, args...)
 end
 function adapt_factors(to, bpc::AbstractBeliefPropagationCache, args...)
   map_factors(adapt(to), bpc, args...)