Skip to content

Commit 37466cc

Browse files
committed
Fix conversion
1 parent 3d5072f commit 37466cc

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

src/optimisation/Optimisation.jl

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -195,11 +195,16 @@ struct OptimLogDensity{
195195
function OptimLogDensity(
196196
model::DynamicPPL.Model,
197197
getlogdensity::Function,
198-
vi::DynamicPPL.VarInfo;
198+
vi::DynamicPPL.AbstractVarInfo;
199199
adtype::ADTypes.AbstractADType=Turing.DEFAULT_ADTYPE,
200200
)
201-
return new{typeof(model),typeof(getlogdensity),typeof(vi),typeof(adtype)}(
202-
DynamicPPL.LogDensityFunction(model, getlogdensity, vi; adtype=adtype)
201+
# Note that typeof(adtype) != typeof(ldf.adtype) in general because of
202+
# DynamicPPL's tweak_adtype
203+
ldf = DynamicPPL.LogDensityFunction(model, getlogdensity, vi; adtype=adtype)
204+
return new{
205+
typeof(ldf.model),typeof(ldf.getlogdensity),typeof(ldf.vi),typeof(ldf.adtype)
206+
}(
207+
ldf
203208
)
204209
end
205210
function OptimLogDensity(

0 commit comments

Comments
 (0)