Python dask.annotate用法及代码示例

用法:
dask.annotate(**annotations)

用于设置 HighLevelGraph 层注释的上下文管理器。

注释是与 dask 调度程序可能选择尊重的任务相关的元数据或软约束：它们在不强制执行硬约束的情况下发出意图。因此，它们主要设计用于分布式调度程序。

几乎任何对象都可以用作注解，但首选 Python 小对象，不鼓励使用大型对象(如 NumPy 数组)。

作为注释提供的可调用对象应采用单个键参数并生成适当的注释。注释集合中的各个任务键被提供给可调用对象。

阵列 A 中的所有任务的优先级应为 100，并在失败时重试 3 次。

>>> import dask
>>> import dask.array as da
>>> with dask.annotate(priority=100, retries=3):
...     A = da.ones((10000, 10000))

将阵列 A 中的任务按扁平块 ID 划分优先级。

>>> nblocks = (10, 10)
>>> with dask.annotate(priority=lambda k: k[1]*nblocks[1] + k[2]):
...     A = da.ones((1000, 1000), chunks=(100, 100))

注释可以嵌套。

>>> with dask.annotate(priority=1):
...     with dask.annotate(retries=3):
...         A = da.ones((1000, 1000))
...     B = A + 1

相关用法

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