Python dask.diagnostics.Callback用法及代码示例

用法: dask.diagnostics.Callback(start=None, start_state=None, pretask=None, posttask=None, finish=None)

使用回调机制的基类

使用以下签名的函数创建回调：

>>> def start(dsk):
...     pass
>>> def start_state(dsk, state):
...     pass
>>> def pretask(key, dsk, state):
...     pass
>>> def posttask(key, result, dsk, state, worker_id):
...     pass
>>> def finish(dsk, state, failed):
...     pass

然后，您可以构造一个包含任意数量的回调对象

>>> cb = Callback(pretask=pretask, finish=finish)

并将其用作计算/获取调用的上下文管理器

>>> with cb:            
...     x.compute()

或使用register 方法全局

>>> cb.register()
>>> cb.unregister()

或者，使用您自己的方法子类化Callback 类。

>>> class PrintKeys(Callback):
...     def _pretask(self, key, dask, state):
...         print("Computing: {0}!".format(repr(key)))

>>> with PrintKeys():   
...     x.compute()

相关用法

注：本文由纯净天空筛选整理自dask.org大神的英文原创作品 dask.diagnostics.Callback。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。