From 82a8338faecc4836f15c7111282db1ebe44a9ca9 Mon Sep 17 00:00:00 2001 From: "houhan@gmail.com" Date: Thu, 21 Dec 2023 06:57:14 +0000 Subject: [PATCH] add stage_task in the diagram --- code/aind_auto_train/plot/curriculum.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/code/aind_auto_train/plot/curriculum.py b/code/aind_auto_train/plot/curriculum.py index aac6432..d0619a8 100644 --- a/code/aind_auto_train/plot/curriculum.py +++ b/code/aind_auto_train/plot/curriculum.py @@ -43,7 +43,7 @@ def draw_diagram_rules(curriculum): # Add nodes (stages) for stage in stages: dot.node(name=stage.name, - label=stage.name, + label=f'{stage.name}\n{curriculum.parameters[stage].task}', shape='ellipse', style='filled', fillcolor=stage_color_mapper[stage.name],