Update SoleDecisionTreeInterface.jl

aclai-lab · Sep 13, 2024 · 0b122cf · 0b122cf
1 parent 9416062
commit 0b122cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/SoleDecisionTreeInterface.jl b/src/SoleDecisionTreeInterface.jl
@@ -79,7 +79,7 @@ function solemodel(tree::DT.Leaf; replace_classlabels = nothing, use_featurename
     return SoleModels.ConstantModel(prediction, info)
 end
 
-function solemodel(model :: DecisionTree.Ensemble)
+function solemodel(model :: DT.Ensemble)
     forest = DecisionForest(map(t -> DecisionTree(solemodel(t)), model.trees))
     return forest
 end