Dask визуализирует несколько выходных узлов на графике Dask - PullRequest
Новая
       35

Dask визуализирует несколько выходных узлов на графике Dask

0 голосов
/ 11 марта 2019

Граф Даска, который я создаю, имеет несколько выходов.

Мне было интересно, можно ли одновременно визуализировать несколько выходов даска.

Когда я пытаюсь использовать dask.visualize(graph). Где graph - это кортеж или словарь узлов Dask. Похоже, что произойдет сбой, создав пустое изображение.

Например: 

import dask

@dask.delayed
def op(x, y):
    return x + y

def graph():
    a = 1
    b = 2
    c = 3
    d = 4

    f = op(a, b)
    g = op(c, d)
    h = op(a, d)

    output1 = op(f, g)
    output2 = op(g, h)

    # works
    dask.visualize(output1, filename='output1.png')
    dask.visualize(output2, filename='output2.png')

    # empty small square box
    dask.visualize((output1, output2), filename='output1and2.png')
    # only outputs the name of the keys
    dask.visualize({'output1': output1, 'output2': output2}, filename='output1and2_.png')

graph()

1 Ответ

1 голос
/ 16 марта 2019

Функция dask.visualize принимает много аргументов, а не последовательность аргументов

Сделайте это 

dask.visualize(output1, output2, filename='output1and2.png')

Вместо этого 

dask.visualize((output1, output2), filename='output1and2.png')
...