elcorto / psweep

Loop like a pro, make parameter studies fun.
https://elcorto.github.io/psweep
BSD 3-Clause "New" or "Revised" License
13 stars 2 forks source link

Use `dask` to launch cluster jobs #9

Closed elcorto closed 9 months ago

elcorto commented 1 year ago

Could be used to replace the custom template file workflow. Neede packages:

dask
dask-jobqueue
distributed
elcorto commented 1 year ago

If this works, we can rename run_local() back to run() and just pass a dask.distributed.Client and use results=client.gather(client.map(worker, params)) instead of results=pool.map(worker, params). Then run() will start batch jobs. The idea behind the name run_local() was to later have a run_cluster() function that would implement a cluster workflow using prep_batch() etc, which would now be unnecessary, since run() can do both.

elcorto commented 10 months ago

This is being worked on in branch https://github.com/elcorto/psweep/tree/feature-dask. Expect frequent rebase and force-push, this is a feature branch :)

So far we have test_dask_local_cluster which works.

Todos

elcorto commented 10 months ago
elcorto commented 9 months ago

Implemented in https://github.com/elcorto/psweep/releases/tag/0.11.0