dask - 如何从 HighLevelGraph 转换为常规 dask dict

Question

我有一些代码，例如：

x = delayed(add)(3, 2)
y = delayed(add)(x, x)

我想得到结果 dask 图：

dsk = {
    'x': (add, 3, 2),
    'y': (add, 'x', 'x')
}

但相反，我得到HighLevelGraph如下：

{'x': {'x': (add, 3, 2)},
 'y': {'y': (add,'x','x')}}

我的问题是：有没有办法轻松地将 HighLevelGraph 字典转换为常规字典？

注意：实际键和可调用对象替换为较短的等效项。

谢谢！

Answer 1

调用dict它。

dsk = dict(dsk)

您可以对任何继承自Mappingdict 类对象的超类的对象执行此操作。