python - 如何可视化 dask 图的子图？

Question

鉴于我有一个非常大的任务图，my_delayed.visualize()要么无法生成，要么太密集而无法在视觉上有用。如果我有特定任务的关键，我可以指定特定深度或 x 数量的父母和孩子来可视化吗？

例如：

from dask import delayed
@delayed
def inc(x):
    return x + 1

def inc_list(x):
    return [inc(n) for n in x]

task = delayed(sum)(inc_list(inc_list([1,2,3])))
task.visualize(n_parents=1)

将生成没有划掉节点的下图。

Answer 1

简短的回答

不，目前不支持此功能

更长的答案

任务图只是一个字典，因此您可以随意操作它，以可视化它的子部分。您可能会查看 dask.core 函数get_dependencies，并get_deps帮助操作图形，以及普通的 Python for 循环。