fix: add missing nout in save_to_file

MilesCranmer · Apr 25, 2024 · 606a7f8 · 606a7f8
1 parent 6690ee3
commit 606a7f8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/SearchUtils.jl b/src/SearchUtils.jl
@@ -406,7 +406,7 @@ Base.@kwdef struct SearchState{
 end
 
 function save_to_file(
-    dominating, j::Integer, dataset::Dataset{T,L}, options::Options
+    dominating, nout::Integer, j::Integer, dataset::Dataset{T,L}, options::Options
 ) where {T,L}
     output_file = options.output_file
     if nout > 1

diff --git a/src/SymbolicRegression.jl b/src/SymbolicRegression.jl
@@ -917,7 +917,7 @@ function _main_search_loop!(
             dominating = calculate_pareto_frontier(state.halls_of_fame[j])
 
             if options.save_to_file
-                save_to_file(dominating, j, dataset, options)
+                save_to_file(dominating, nout, j, dataset, options)
             end
             ###################################################################
             # Migration #######################################################