mindee / tawazi

A DAG Scheduler library written in pure python
https://mindee.github.io/tawazi/
Apache License 2.0
83 stars 5 forks source link

chg: enhance draw utility #214

Closed WassimMindee closed 5 months ago

WassimMindee commented 6 months ago

Description

Reworking draw feature in tawazi for interactive display of DAGs

Example:

@dag(max_concurrency=2)
def pipeline():
    var_node1 = node1()
    var_node2 = node2()
    var_node6 = node6()
    var_node3 = node3(var_node1, var_node2)
    var_node4 = node4(var_node3, var_node2)
    var_node5 = node5(var_node4, var_node6)

    return var_node5, var_node3
pipeline.draw(fig_name="my_pipeline", show_nodes_names=True)

image

Fixes # (issue)

Type of change

Please delete options that are not relevant.

How Has This Been Tested?

Please describe the tests that you ran to verify your changes. Provide instructions so we can reproduce. Please also list any relevant details for your test configuration

Checklist: