Only continuous input allowed

MilesCranmer · Jul 4, 2023 · 78bc164 · 78bc164
1 parent 1ea518f
commit 78bc164
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/src/MLJInterface.jl b/src/MLJInterface.jl
@@ -166,13 +166,8 @@ MMI.package_license(::Type{<:AbstractSRRegressor}) = "Apache-2.0"
 MMI.is_pure_julia(::Type{<:AbstractSRRegressor}) = true
 MMI.is_wrapper(::Type{<:AbstractSRRegressor}) = false
 
-MMI.input_scitype(::Type{SRRegressor}) = MMI.Table(Union{MMI.Continuous,MMI.Count})
-function MMI.target_scitype(::Type{SRRegressor})
-    return Union{
-        MMI.Table(Union{MMI.Continuous,MMI.Count}),
-        AbstractVector{<:Union{MMI.Continuous,MMI.Count}},
-    }
-end
+MMI.input_scitype(::Type{SRRegressor}) = MMI.Table(MMI.Continuous)
+MMI.target_scitype(::Type{SRRegressor}) = Union{MMI.Table(MMI.Continuous),AbstractVector{<:MMI.Continuous}}
 MMI.supports_weights(::Type{SRRegressor}) = true
 MMI.human_name(::Type{SRRegressor}) = "Symbolic Regression via Evolutionary Search"
 MMI.load_path(::Type{SRRegressor}) = "SymbolicRegression.MLJInterfaceModule.SRRegressor"