Skip to content

Commit

Permalink
fix: MLJ with logger
Browse files Browse the repository at this point in the history
  • Loading branch information
MilesCranmer committed Nov 8, 2024
1 parent bce1df5 commit 87cdffe
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions src/MLJInterface.jl
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ using ..CoreModule.OptionsModule: DEFAULT_OPTIONS, OPTION_DESCRIPTIONS
using ..ComplexityModule: compute_complexity
using ..HallOfFameModule: HallOfFame, format_hall_of_fame
using ..UtilsModule: subscriptify, @ignore
using ..LoggingModule: AbstractSRLogger

import ..equation_search

Expand All @@ -57,9 +58,7 @@ function modelexpr(model_name::Symbol)
procs::Union{Vector{Int},Nothing} = nothing
addprocs_function::Union{Function,Nothing} = nothing
heap_size_hint_in_bytes::Union{Integer,Nothing} = nothing
logger::Union{AbstractLogger,Nothing} = nothing
logging_callback::Union{Function,Nothing} = nothing
log_every_n::Union{Integer,NamedTuple} = 1
logger::Union{AbstractSRLogger,Nothing} = nothing
runtests::Bool = true
run_id::Union{String,Nothing} = nothing
loss_type::L = Nothing
Expand Down Expand Up @@ -270,8 +269,6 @@ function _update(
verbosity=verbosity,
extra=isnothing(class) ? (;) : (; class),
logger=m.logger,
logging_callback=m.logging_callback,
log_every_n=m.log_every_n,
# Help out with inference:
v_dim_out=isa(m, SRRegressor) ? Val(1) : Val(2),
)
Expand Down

0 comments on commit 87cdffe

Please sign in to comment.