Skip to content

Commit 307a243

Browse files
committed
trying to sort out issue with laplace
1 parent 09848de commit 307a243

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/models/laplace.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ end
1313

1414
using Flux
1515
using Flux.Optimise: update!
16-
using CounterfactualExplanations
16+
using CounterfactualExplanations.Models: LaplaceReduxModel
1717
"""
1818
train(M::LaplaceModel, data::CounterfactualData; kwargs...)
1919
@@ -38,7 +38,7 @@ function train(M::LaplaceReduxModel, data::CounterfactualData; kwargs...)
3838
# Fit Laplace:
3939
la = Laplace(model, λ=args.λ, H₀=args.H₀)
4040
LaplaceRedux.fit!(la, data)
41-
M = LaplaceReduxModel(la)
41+
M = CounterfactualExplanations.Models.LaplaceReduxModel(la)
4242

4343
return M
4444

0 commit comments

Comments
 (0)