[Fea] Run function on all workers within an annotate context
Open
- Dominant language
- Python
- Stars
- 1.7k
- Forks
- 778
- Avg merge
- 2h 50m
- Merged PRs (30d)
- 3
Description
`client.run` allows users to pass a list of workers to run a function on specific workers. It would be good if `client.run` could be used in conjunction with the annotate context to run on all workers with a specific resource tag.
Eg:
```python
with dask.annotate(resources={'GPU': 1}):
client.run(gpu_init_func)
```
cc: @jakirkham @beckernick
Contributor guide
Assessment
This issue has not been assessed yet.