如何可视化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循环。