Spaces:
Sleeping
Sleeping
AutonLabTruth
commited on
Merge branch 'recover' of https://github.com/DhananjayAshok/PySR into recover
Browse files- pysr/sr.py +3 -2
pysr/sr.py
CHANGED
@@ -251,7 +251,8 @@ def pysr(X=None, y=None, weights=None,
|
|
251 |
weightMutateOperator=weightMutateOperator,
|
252 |
weightRandomize=weightRandomize,
|
253 |
weightSimplify=weightSimplify,
|
254 |
-
constraints=constraints
|
|
|
255 |
|
256 |
kwargs = {**_set_paths(tempdir), **kwargs}
|
257 |
|
@@ -301,7 +302,7 @@ def _final_pysr_process(julia_optimization, procs, runfile_filename, timeout, **
|
|
301 |
if timeout is not None:
|
302 |
command = [f'timeout', f'{timeout}'] + command
|
303 |
print("Running on", ' '.join(command))
|
304 |
-
process = subprocess.Popen(command, stdout=subprocess.PIPE, bufsize=1
|
305 |
try:
|
306 |
while True:
|
307 |
line = process.stdout.readline()
|
|
|
251 |
weightMutateOperator=weightMutateOperator,
|
252 |
weightRandomize=weightRandomize,
|
253 |
weightSimplify=weightSimplify,
|
254 |
+
constraints=constraints,
|
255 |
+
extra_sympy_mappings=extra_sympy_mappings)
|
256 |
|
257 |
kwargs = {**_set_paths(tempdir), **kwargs}
|
258 |
|
|
|
302 |
if timeout is not None:
|
303 |
command = [f'timeout', f'{timeout}'] + command
|
304 |
print("Running on", ' '.join(command))
|
305 |
+
process = subprocess.Popen(command, stdout=subprocess.PIPE, bufsize=1)
|
306 |
try:
|
307 |
while True:
|
308 |
line = process.stdout.readline()
|