AutonLabTruth commited on
Commit
5f43445
2 Parent(s): aebcf1a f5cd852

Merge branch 'recover' of https://github.com/DhananjayAshok/PySR into recover

Browse files
Files changed (1) hide show
  1. 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, shell=True)
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()