Spaces:
Running
Running
MilesCranmer
commited on
Commit
•
8d92d8e
1
Parent(s):
4ef59ef
Correctly allow for deleted hall of fame
Browse files- 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
|
|
|
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):
|