fix: return gradlog as vector

ReactiveBayes · Dec 15, 2023 · 5f849dc · 5f849dc
1 parent f274500
commit 5f849dc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/distributions/poisson.jl b/src/distributions/poisson.jl
@@ -61,7 +61,8 @@ getlogpartition(::NaturalParametersSpace, ::Type{Poisson}) = (η) -> begin
 end
 
 getgradlogpartition(::NaturalParametersSpace, ::Type{Poisson}) = (η) -> begin
-    return getlogpartition(η, Poisson)
+    (η1,) = unpack_parameters(Poisson, η)
+    return SA[exp(η1)]
 end
 
 getfisherinformation(::NaturalParametersSpace, ::Type{Poisson}) = (η) -> begin