From 5111a878f6be39b03a1a02d3b57e7065d2d33337 Mon Sep 17 00:00:00 2001 From: dehann Date: Tue, 13 Aug 2024 01:00:29 -0700 Subject: [PATCH] bug fix unpackDistr PackedZeroMeanFullNormal --- src/Serialization/services/SerializingDistributions.jl | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/Serialization/services/SerializingDistributions.jl b/src/Serialization/services/SerializingDistributions.jl index 09b25631..7fb7e58f 100644 --- a/src/Serialization/services/SerializingDistributions.jl +++ b/src/Serialization/services/SerializingDistributions.jl @@ -24,7 +24,8 @@ function unpackDistribution(dtr::PackedZeroMeanDiagNormal) return MvNormal(LinearAlgebra.Diagonal(map(abs2, sqrt.(dtr.diag)))) end # sqrt.(dtr.diag) function unpackDistribution(dtr::PackedZeroMeanFullNormal) - return MvNormal(reshape(dtr.cov, length(dtr.mu), :)) + d = round(Int,sqrt(size(dtr.cov)[1])) + return MvNormal(reshape(dtr.cov, d, d)) end unpackDistribution(dtr::PackedDiagNormal) = MvNormal(dtr.mu, sqrt.(dtr.diag)) function unpackDistribution(dtr::PackedFullNormal)