Skip to content

Commit 9942b52

Browse files
committed
Fix DecisionSet"
1 parent 58e808f commit 9942b52

File tree

2 files changed

+15
-0
lines changed

2 files changed

+15
-0
lines changed

src/SoleModels.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ using Reexport
44
using StatsBase
55
using ThreadSafeDicts
66
using Lazy
7+
using Lazy: @forward
78
using FunctionWrappers: FunctionWrapper
89
using CategoricalArrays
910

@@ -59,6 +60,7 @@ export height
5960

6061
export DecisionEnsemble, models
6162
export DecisionForest, trees
63+
export DecisionSet, rules, nrules
6264

6365
export MixedModel
6466

src/utils/models/other.jl

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -344,6 +344,19 @@ end
344344
rules(m::DecisionSet) = m.rules
345345
nrules(m::DecisionSet) = length(rules(m))
346346

347+
# Helpers
348+
@forward DecisionSet.rules (
349+
Base.length,
350+
Base.getindex,
351+
Base.setindex!,
352+
Base.push!,
353+
Base.pushfirst!,
354+
Base.append!,
355+
Base.iterate, Base.IteratorSize, Base.IteratorEltype,
356+
Base.firstindex, Base.lastindex,
357+
Base.keys, Base.values,
358+
)
359+
347360
iscomplete(m::DecisionSet) = m.iscomplete
348361
isnonoverlapping(m::DecisionSet) = m.isnonoverlapping
349362

0 commit comments

Comments
 (0)