mars.contrib.dask.mars_scheduler#

mars.contrib.dask.mars_scheduler(dsk: dict, keys: Union[List[List[str]], List[str]])[source]#

A Dask-Mars scheduler

This scheduler is intended to be compatible with existing dask user interface, no callbacks are implemented.

Parameters
  • dsk (Dict) – Dask graph, represented as a task DAG dictionary.

  • keys (Union[List[List[str]], List[str]]) – 1d or 2d list of Dask graph keys whose values we wish to compute and return.

Returns

Computed values corresponding to the provided keys with same dimension.

Return type

Object