Pass arguments to dask.compute()

[crusaderky]

I work with a very large dask-based algorithm in xarray, and I do my optimization by hand before hitting compute(). In other cases, I need using multiple dask schedulers at once (e.g. a multithreaded one for numpy-based work and a multiprocessing one for pure python work).

This change proposal (which I'm happy to do) is about accepting *args, **kwds parameters in all .compute(), .load(), and .persist() xarray methods and pass them verbatim to the underlying dask compute() and persist() functions.

[shoyer]

Yes, this sounds like a great idea to me!

[crusaderky]

Working on it now

[jhamman]

@crusaderky - did you make any progress here? I also have a feature branch that I started on the plane a few days back...

[crusaderky]

@jhamman I finished the implementation I have to write the unit tests this weekend

[crusaderky]

@jhamman 's implementation is far more complete than mine, I pass the trophy :)