mars.contrib.dask.mars_scheduler

mars.contrib.dask.mars_scheduler(dsk: dict, keys: List[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 (List[List[str]]) – 2d-list of Dask graph keys whose values we wish to compute and return.

Returns

Computed values corresponding to the provided keys.

Return type

Object