MilesCranmer commited on
Commit
964b669
1 Parent(s): fef663e

Reset global equation file name

Browse files
Files changed (1) hide show
  1. pysr/sr.py +5 -0
pysr/sr.py CHANGED
@@ -358,6 +358,11 @@ def get_hof(equation_file=None, n_features=None, variable_names=None, extra_symp
358
  if variable_names is None: variable_names = global_variable_names
359
  if extra_sympy_mappings is None: extra_sympy_mappings = global_extra_sympy_mappings
360
 
 
 
 
 
 
361
  try:
362
  output = pd.read_csv(equation_file + '.bkup', sep="|")
363
  except FileNotFoundError:
 
358
  if variable_names is None: variable_names = global_variable_names
359
  if extra_sympy_mappings is None: extra_sympy_mappings = global_extra_sympy_mappings
360
 
361
+ global_equation_file = equation_file
362
+ global_n_features = n_features
363
+ global_variable_names = variable_names
364
+ global_extra_sympy_mappings = extra_sympy_mappings
365
+
366
  try:
367
  output = pd.read_csv(equation_file + '.bkup', sep="|")
368
  except FileNotFoundError: