ray.util.dask.RayDaskCallback#

class ray.util.dask.RayDaskCallback(*args: Any, **kwargs: Any)[源代码]#

基类：Callback

扩展了 Dask 的 Callback 类，增加了 Ray 特定的钩子。在实例化或子类化此类时，可以提供正常的 Dask 钩子（例如 pretask、posttask 等）和 Ray 特定的钩子。

参见 dask.callbacks.Callback 以了解用法。

注意事项：任何 Dask-Ray 调度器必须使用 local_ray_callbacks 上下文管理器将 Ray 特定的回调引入上下文中，因为 Dask 提供的内置 local_callbacks 上下文管理器并不知道这个类。

方法

属性