ray.util.dask.callbacks.RayDaskCallback._ray_postsubmit#

RayDaskCallback._ray_postsubmit(task, key, deps, object_ref: ray._raylet.ObjectRef)[source]#

Run after submitting a Ray task.

Parameters:

  • task – A Dask task, where the first tuple item is the task function, and the remaining tuple items are the task arguments, which are either the actual argument values, or Dask keys into the deps dictionary whose corresponding values are the argument values.

  • key – The Dask graph key for the given task.

  • deps – The dependencies of this task.

  • object_ref – The object reference for the return value of the Ray task.