Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Use dask to launch cluster jobs #9

Closed
elcorto opened this issue Jan 16, 2023 · 4 comments
Closed

Use dask to launch cluster jobs #9

elcorto opened this issue Jan 16, 2023 · 4 comments
Assignees

Comments

@elcorto
Copy link
Owner

elcorto commented Jan 16, 2023

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

dask
dask-jobqueue
distributed
@elcorto elcorto self-assigned this Jan 16, 2023
@elcorto
Copy link
Owner Author

elcorto commented Jan 21, 2023

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
Copy link
Owner Author

elcorto commented Sep 29, 2023

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

  • split batch_with_git example into batch_templates (w/o git) and batch_templates_with_git
    • make sure that test_all.py::test_run_all_examples pickes up all of them
  • Add small example using SLURMCluster (for now in the manual)
  • Add example explaining all parameters of SLURMCluster
  • Update manual (partially done)
  • CI: install dask in addition to test reqs

@elcorto
Copy link
Owner Author

elcorto commented Oct 29, 2023

@elcorto elcorto closed this as completed Oct 29, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant