MilesCranmer commited on
Commit
fb2f513
1 Parent(s): 893fdd2

Do not store custom loss in Julia main

Browse files
Files changed (1) hide show
  1. pysr/sr.py +2 -2
pysr/sr.py CHANGED
@@ -1307,7 +1307,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
1307
  complexity_of_operators_str += ")"
1308
  complexity_of_operators = Main.eval(complexity_of_operators_str)
1309
 
1310
- Main.custom_loss = Main.eval(self.loss)
1311
 
1312
  mutationWeights = [
1313
  float(self.weight_mutate_constant),
@@ -1331,7 +1331,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
1331
  complexity_of_constants=self.complexity_of_constants,
1332
  complexity_of_variables=self.complexity_of_variables,
1333
  nested_constraints=nested_constraints,
1334
- loss=Main.custom_loss,
1335
  maxsize=int(self.maxsize),
1336
  hofFile=_escape_filename(self.equation_file_),
1337
  npopulations=int(self.populations),
 
1307
  complexity_of_operators_str += ")"
1308
  complexity_of_operators = Main.eval(complexity_of_operators_str)
1309
 
1310
+ custom_loss = Main.eval(self.loss)
1311
 
1312
  mutationWeights = [
1313
  float(self.weight_mutate_constant),
 
1331
  complexity_of_constants=self.complexity_of_constants,
1332
  complexity_of_variables=self.complexity_of_variables,
1333
  nested_constraints=nested_constraints,
1334
+ loss=custom_loss,
1335
  maxsize=int(self.maxsize),
1336
  hofFile=_escape_filename(self.equation_file_),
1337
  npopulations=int(self.populations),