Fix caltable bug

ptiede · Oct 16, 2023 · cf5cfb1 · cf5cfb1
1 parent 70bc674
commit cf5cfb1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/bayes/bayes.jl b/src/bayes/bayes.jl
@@ -49,6 +49,8 @@ function DensityInterface.logdensityof(post::Posterior, x)
     return logdensityof(post.lklhd, x) + pr
 end
 
+(post::Posterior)(θ) = logdensityof(post, θ)
+
 """
     prior_sample([rng::AbstractRandom], post::Posterior, args...)
 
@@ -137,6 +139,9 @@ struct TransformedPosterior{P<:Posterior,T} <: AbstractPosterior
     transform::T
 end
 
+(tpost::TransformedPosterior)(θ) = logdensityof(tpost, θ)
+
+
 function prior_sample(rng, tpost::TransformedPosterior, args...)
     inv = Base.Fix1(HypercubeTransform.inverse, tpost)
     map(inv, prior_sample(rng, tpost.lpost, args...))

diff --git a/src/calibration/caltable.jl b/src/calibration/caltable.jl
@@ -250,7 +250,8 @@ end
 function stations(g::JonesCache)
     s1 = keys(g.seg)
     if !(g.references isa AbstractVector{<:NoReference})
-        return sort(unique(vcat(s1, getproperty.(g.references, :site))))
+        s = (s1..., getproperty.(g.references, :site)...)
+        return sort(unique(s))
     end
     return sort(unique(s1))
 end