tidy up

Author: sunxd3

test/gibbs_example/gmm.jl

@@ -38,7 +38,7 @@ function log_joint(; μ, w, z, x)
     return logp
 end
 
-function condition(gmm::GMM, conditioned_values::NamedTuple)
+function AbstractMCMC.condition(gmm::GMM, conditioned_values::NamedTuple)
     return ConditionedGMM(gmm.data, conditioned_values)
 end
 
@@ -69,11 +69,3 @@ end
 function LogDensityProblems.capabilities(::ConditionedGMM)
     return LogDensityProblems.LogDensityOrder{0}()
 end
-
-function flatten(nt::NamedTuple)
-    return only(values(nt))
-end
-
-function unflatten(vec::AbstractVector, group::Tuple)
-    return NamedTuple((only(group) => vec,))
-end

test/gibbs_example/hier_normal.jl

@@ -57,7 +57,3 @@ end
 function LogDensityProblems.capabilities(::ConditionedHierNormal)
     return LogDensityProblems.LogDensityOrder{0}()
 end
-
-function AbstractMCMC.recompute_logprob!!(hn::ConditionedHierNormal, vals, state)
-    return AbstractMCMC.set_logprob!!(state, LogDensityProblems.logdensity(hn, vals))
-end