set MMI is_wrapper trait as true (#22)

JuliaAI · Jul 28, 2024 · 466ea27 · 466ea27
1 parent e0a8ebd
commit 466ea27
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/models/rfe.jl b/src/models/rfe.jl
@@ -397,6 +397,7 @@ end
 MMI.load_path(::Type{<:RFE}) = "FeatureSelection.RecursiveFeatureElimination"
 MMI.constructor(::Type{<:RFE}) = RecursiveFeatureElimination
 MMI.package_name(::Type{<:RFE}) = "FeatureSelection"
+MMI.is_wrapper(::Type{<:RFE}) = true
 
 for trait in [
     :supports_weights,