Автоматизировать создание графиков высокого уровня в dask

Question

Я пытаюсь автоматизировать создание HighLevelGraphs в dask.

пример
при запуске графика, который я ввел вручную, график может быть обработан без ошибок

созданный вручную график (псевдокод):

import time
from dask.highlevelgraph import HighLevelGraph

def func(x):
    time.sleep(30)
    pass

layers = {'test': {'test_1': (func, x) }
        .............}

dependencies = ............

graph = HighLevelGraph(layers, dependencies)

Для автоматизации построения графиков я создал функцию, которая создает словарь и принимает параметр processing_function, которому я передаю func слой становится:

layers = {'test': {'test_1': (<function func(x)>, x) }
        .............}

и я не могу запустить график

Вопрос
Что я должен передать в параметры processing_function, чтобы получить func а не <function func(x)>

Большое спасибо за помощь! Оставайтесь в безопасности!