Skip to content

Commit f758a4c

Browse files
Apply suggestions from code review
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
1 parent 67ff8e8 commit f758a4c

File tree

1 file changed

+12
-4
lines changed

1 file changed

+12
-4
lines changed

gibbs_example/gmm.jl

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -44,13 +44,21 @@ function condition(gmm::GMM, conditioned_values::NamedTuple)
4444
return ConditionedGMM(gmm.data, conditioned_values)
4545
end
4646

47-
function LogDensityProblems.logdensity(gmm::ConditionedGMM{names}, params::AbstractVector) where {names}
47+
function LogDensityProblems.logdensity(
48+
gmm::ConditionedGMM{names}, params::AbstractVector
49+
) where {names}
4850
if Set(names) == Set([, :w]) # conditioned on μ, w, so params are z
49-
return log_joint(; μ=gmm.conditioned_values.μ, w=gmm.conditioned_values.w, z=params, x=gmm.data.x)
51+
return log_joint(;
52+
μ=gmm.conditioned_values.μ, w=gmm.conditioned_values.w, z=params, x=gmm.data.x
53+
)
5054
elseif Set(names) == Set([:z, :w]) # conditioned on z, w, so params are μ
51-
return log_joint(; μ=params, w=gmm.conditioned_values.w, z=gmm.conditioned_values.z, x=gmm.data.x)
55+
return log_joint(;
56+
μ=params, w=gmm.conditioned_values.w, z=gmm.conditioned_values.z, x=gmm.data.x
57+
)
5258
elseif Set(names) == Set([:z, ]) # conditioned on z, μ, so params are w
53-
return log_joint(; μ=gmm.conditioned_values.μ, w=params, z=gmm.conditioned_values.z, x=gmm.data.x)
59+
return log_joint(;
60+
μ=gmm.conditioned_values.μ, w=params, z=gmm.conditioned_values.z, x=gmm.data.x
61+
)
5462
else
5563
error("Unsupported conditioning configuration.")
5664
end

0 commit comments

Comments
 (0)