Skip to content

Commit 015534b

Browse files
committed
fix comma -> semicolon typo
1 parent 06599cf commit 015534b

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

ext/DynamicPPLMarginalLogDensitiesExt.jl

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,6 @@ module DynamicPPLMarginalLogDensitiesExt
33
using DynamicPPL: DynamicPPL, LogDensityProblems, VarName
44
using MarginalLogDensities: MarginalLogDensities
55

6-
_to_varname(n::Symbol) = VarName{n}()
7-
_to_varname(n::VarName) = n
8-
96
# A thin wrapper to adapt a DynamicPPL.LogDensityFunction to the interface expected by
107
# MarginalLogDensities. It's helpful to have a struct so that we can dispatch on its type
118
# below.
@@ -19,7 +16,7 @@ end
1916
"""
2017
marginalize(
2118
model::DynamicPPL.Model,
22-
varnames::AbstractVector{<:Union{Symbol,<:VarName}};
19+
marginalized_varnames::AbstractVector{<:VarName};
2320
varinfo::DynamicPPL.AbstractVarInfo=link(VarInfo(model), model),
2421
getlogprob=DynamicPPL.getlogjoint,
2522
method::MarginalLogDensities.AbstractMarginalizer=MarginalLogDensities.LaplaceApprox();
@@ -93,15 +90,14 @@ julia> logpdf(Normal(2.0), 1.0)
9390
"""
9491
function DynamicPPL.marginalize(
9592
model::DynamicPPL.Model,
96-
varnames::AbstractVector{<:Union{Symbol,<:VarName}};
93+
marginalized_varnames::AbstractVector{<:VarName};
9794
varinfo::DynamicPPL.AbstractVarInfo=DynamicPPL.link(DynamicPPL.VarInfo(model), model),
9895
getlogprob::Function=DynamicPPL.getlogjoint,
9996
method::MarginalLogDensities.AbstractMarginalizer=MarginalLogDensities.LaplaceApprox(),
10097
kwargs...,
10198
)
10299
# Determine the indices for the variables to marginalise out.
103-
vns = map(_to_varname, varnames)
104-
varindices = reduce(vcat, DynamicPPL.vector_getranges(varinfo, vns))
100+
varindices = reduce(vcat, DynamicPPL.vector_getranges(varinfo, marginalized_varnames))
105101
# Construct the marginal log-density model.
106102
f = DynamicPPL.LogDensityFunction(model, getlogprob, varinfo)
107103
mld = MarginalLogDensities.MarginalLogDensity(

0 commit comments

Comments
 (0)