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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。