Specialize on options again

src/MLJInterface.jl

@@ -39,6 +39,7 @@ function modelexpr(model_name::Symbol)
             numprocs::Union{Int,Nothing} = nothing
             procs::Union{Vector{Int},Nothing} = nothing
             addprocs_function::Union{Function,Nothing} = nothing
+            heap_size_hint_in_bytes::Union{Integer,Nothing} = nothing
             runtests::Bool = true
             loss_type::L = Nothing
             selection_method::Function = choose_best
@@ -166,6 +167,7 @@ function _update(m, verbosity, old_fitresult, old_cache, X, y, w, options)
         numprocs=m.numprocs,
         procs=m.procs,
         addprocs_function=m.addprocs_function,
+        heap_size_hint_in_bytes=m.heap_size_hint_in_bytes,
         runtests=m.runtests,
         saved_state=(old_fitresult === nothing ? nothing : old_fitresult.state),
         return_state=true,

src/SymbolicRegression.jl

@@ -422,7 +422,7 @@ end
 function equation_search(
     datasets::Vector{D};
     niterations::Int=10,
-    @nospecialize(options::Options = Options()),
+    options::Options = Options(),
     parallelism=:multithreading,
     numprocs::Union{Int,Nothing}=nothing,
     procs::Union{Vector{Int},Nothing}=nothing,
@@ -529,6 +529,7 @@ function equation_search(
         )
         heap_size_hint_in_bytes === nothing &&
             @info "Automatically setting --heap-size-hint=$(heap_size_hint_in_megabytes)M on each Julia process. You can set this manually with `heap_size_hint_in_bytes`."
+
         `--heap-size=$(heap_size_hint_in_megabytes)M`
     else
         ``
@@ -558,7 +559,7 @@ function _equation_search(
     ::Val{dim_out},
     datasets::Vector{D},
     niterations::Int,
-    @nospecialize(options::Options),
+    options::Options,
     numprocs::Integer,
     procs::Union{Vector{Int},Nothing},
     addprocs_function::F,