MilesCranmer commited on
Commit
7e735f6
1 Parent(s): ecc6ae8

Define run function on all workers

Browse files
Files changed (1) hide show
  1. pysr/sr.py +1 -1
pysr/sr.py CHANGED
@@ -197,7 +197,7 @@ const weights = convert(Array{Float32, 1}, """f"{weight_str})"
197
  with open(f'/tmp/.runfile_{rand_string}.jl', 'w') as f:
198
  print(f'@everywhere include("/tmp/.hyperparams_{rand_string}.jl")', file=f)
199
  print(f'@everywhere include("/tmp/.dataset_{rand_string}.jl")', file=f)
200
- print(f'include("{pkg_directory}/sr.jl")', file=f)
201
  print(f'fullRun({niterations:d}, npop={npop:d}, ncyclesperiteration={ncyclesperiteration:d}, fractionReplaced={fractionReplaced:f}f0, verbosity=round(Int32, {verbosity:f}), topn={topn:d})', file=f)
202
  print(f'rmprocs(nprocs)', file=f)
203
 
 
197
  with open(f'/tmp/.runfile_{rand_string}.jl', 'w') as f:
198
  print(f'@everywhere include("/tmp/.hyperparams_{rand_string}.jl")', file=f)
199
  print(f'@everywhere include("/tmp/.dataset_{rand_string}.jl")', file=f)
200
+ print(f'@everywhere include("{pkg_directory}/sr.jl")', file=f)
201
  print(f'fullRun({niterations:d}, npop={npop:d}, ncyclesperiteration={ncyclesperiteration:d}, fractionReplaced={fractionReplaced:f}f0, verbosity=round(Int32, {verbosity:f}), topn={topn:d})', file=f)
202
  print(f'rmprocs(nprocs)', file=f)
203