Skip to content

Commit 1b73e5a

Browse files
committed
don't think it really needs those type params
1 parent 37466cc commit 1b73e5a

File tree

1 file changed

+3
-11
lines changed

1 file changed

+3
-11
lines changed

src/optimisation/Optimisation.jl

Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -187,25 +187,17 @@ optim_ld = OptimLogDensity(model, varinfo)
187187
optim_ld(z) # returns -logp
188188
```
189189
"""
190-
struct OptimLogDensity{
191-
M<:DynamicPPL.Model,F<:Function,V<:DynamicPPL.AbstractVarInfo,AD<:ADTypes.AbstractADType
192-
}
193-
ldf::DynamicPPL.LogDensityFunction{M,F,V,AD}
190+
struct OptimLogDensity{L<:DynamicPPL.LogDensityFunction}
191+
ldf::L
194192

195193
function OptimLogDensity(
196194
model::DynamicPPL.Model,
197195
getlogdensity::Function,
198196
vi::DynamicPPL.AbstractVarInfo;
199197
adtype::ADTypes.AbstractADType=Turing.DEFAULT_ADTYPE,
200198
)
201-
# Note that typeof(adtype) != typeof(ldf.adtype) in general because of
202-
# DynamicPPL's tweak_adtype
203199
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
208-
)
200+
return new{typeof(ldf)}(ldf)
209201
end
210202
function OptimLogDensity(
211203
model::DynamicPPL.Model,

0 commit comments

Comments
 (0)