Only continuous input allowed

MilesCranmer · Jul 4, 2023 · dbf540b · dbf540b
1 parent 1ea518f
commit dbf540b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 8 deletions.
diff --git a/src/MLJInterface.jl b/src/MLJInterface.jl
@@ -166,12 +166,9 @@ 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})
+MMI.input_scitype(::Type{SRRegressor}) = MMI.Table(MMI.Continuous)
 function MMI.target_scitype(::Type{SRRegressor})
-    return Union{
-        MMI.Table(Union{MMI.Continuous,MMI.Count}),
-        AbstractVector{<:Union{MMI.Continuous,MMI.Count}},
-    }
+    return Union{MMI.Table(MMI.Continuous),AbstractVector{<:MMI.Continuous}}
 end
 MMI.supports_weights(::Type{SRRegressor}) = true
 MMI.human_name(::Type{SRRegressor}) = "Symbolic Regression via Evolutionary Search"

diff --git a/src/Options.jl b/src/Options.jl
@@ -364,7 +364,7 @@ function Options end
     una_constraints=nothing,
     progress::Bool=true,
     terminal_width::Union{Nothing,Integer}=nothing,
-    optimizer_algorithm::String="BFGS",
+    optimizer_algorithm::AbstractString="BFGS",
     optimizer_nrestarts::Integer=2,
     optimizer_probability::Real=0.14,
     optimizer_iterations::Union{Nothing,Integer}=nothing,

diff --git a/src/OptionsStruct.jl b/src/OptionsStruct.jl
@@ -201,8 +201,8 @@ function Base.print(io::IO, options::Options)
                 fieldname in fieldnames(Options) if fieldname ∉ [:operators, :nuna, :nbin]
             ],
             ", ",
-        )
-        * ")",
+        ) *
+        ")",
     )
 end
 Base.show(io::IO, ::MIME"text/plain", options::Options) = Base.print(io, options)