fix: grad is a vector, not a scalar

ReactiveBayes · Dec 15, 2023 · 9ad7b0a · 9ad7b0a
1 parent f05cfd3
commit 9ad7b0a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/distributions/exponential.jl b/src/distributions/exponential.jl
@@ -46,7 +46,7 @@ end
 
 getgradlogpartition(::NaturalParametersSpace, ::Type{Exponential}) = (η) -> begin
     (η₁,) = unpack_parameters(Exponential, η)
-    return -1/η₁
+    return SA[-1/η₁]
 end
 
 getfisherinformation(::NaturalParametersSpace, ::Type{Exponential}) = (η) -> begin