MilesCranmer commited on
Commit
8d92d8e
1 Parent(s): 4ef59ef

Correctly allow for deleted hall of fame

Browse files
Files changed (1) hide show
  1. pysr/sr.py +4 -1
pysr/sr.py CHANGED
@@ -313,10 +313,13 @@ def pysr(X=None, y=None, weights=None,
313
  _final_pysr_process(**kwargs)
314
  _set_globals(**kwargs)
315
 
 
 
316
  if delete_tempfiles:
317
  shutil.rmtree(kwargs['tmpdir'])
318
 
319
- return get_hof(**kwargs)
 
320
 
321
 
322
  def _set_globals(X, equation_file, extra_sympy_mappings, variable_names, **kwargs):
 
313
  _final_pysr_process(**kwargs)
314
  _set_globals(**kwargs)
315
 
316
+ equations = get_hof(**kwargs)
317
+
318
  if delete_tempfiles:
319
  shutil.rmtree(kwargs['tmpdir'])
320
 
321
+ return equations
322
+
323
 
324
 
325
  def _set_globals(X, equation_file, extra_sympy_mappings, variable_names, **kwargs):