Closed andylizf closed 2 weeks ago
Resolve #4277.
Adds a plot method to the Dag class to visualize task dependencies. Automatically displays inline in Jupyter or saves to a file in other environments.
plot
Dag
from sky import Dag, Task with Dag() as dag: task_a = Task(name="A") task_b = Task(name="B") task_c = Task(name="C") task_a >> task_b >> task_c dag.plot()
Notes:
matplotlib
networkx
Tested (run the relevant ones):
bash format.sh
.ipynb
pytest tests/test_smoke.py
pytest tests/test_smoke.py::test_fill_in_the_name
conda deactivate; bash -i tests/backward_compatibility_tests.sh
@cblmemo PTAL, thanks!
Hi @cblmemo, could you handle the merge for me?
Resolve #4277.
Adds a
plot
method to theDag
class to visualize task dependencies. Automatically displays inline in Jupyter or saves to a file in other environments.Example:
Notes:
matplotlib
andnetworkx
.Tested (run the relevant ones):
bash format.sh
.ipynb
)pytest tests/test_smoke.py
pytest tests/test_smoke.py::test_fill_in_the_name
conda deactivate; bash -i tests/backward_compatibility_tests.sh