Update deprecated varMap

MilesCranmer · Jun 4, 2023 · eb0a059 · eb0a059
1 parent 530e565
commit eb0a059
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/src/SearchUtils.jl b/src/SearchUtils.jl
@@ -314,13 +314,13 @@ end
 
 load_saved_population(::Nothing; kws...) = nothing
 
-function make_datasets(X, y, weights, varMap, loss_type)
+function make_datasets(X, y, weights, variable_names, loss_type)
     nout = size(y, FEATURE_DIM)
     first_dataset = Dataset(
         X,
         y[1, :];
         weights=(weights === nothing ? weights : weights[1, :]),
-        varMap=varMap,
+        variable_names=variable_names,
         loss_type=loss_type,
     )
     datasets = Vector{typeof(first_dataset)}()
@@ -333,7 +333,7 @@ function make_datasets(X, y, weights, varMap, loss_type)
                     X,
                     y[j, :];
                     weights=(weights === nothing ? weights : weights[j, :]),
-                    varMap=varMap,
+                    variable_names=variable_names,
                     loss_type=loss_type,
                 )::typeof(first_dataset),
             )

diff --git a/src/SymbolicRegression.jl b/src/SymbolicRegression.jl
@@ -316,7 +316,6 @@ function EquationSearch(
     end
     variable_names = deprecate_varmap(variable_names, varMap, :EquationSearch)
 
-    nout = size(y, FEATURE_DIM)
     if weights !== nothing
         weights = reshape(weights, size(y))
     end