From 01cc13f495decbd6f8f716d7be586e02864f3434 Mon Sep 17 00:00:00 2001 From: Kento Mori Date: Fri, 18 Oct 2019 05:24:55 +0900 Subject: [PATCH] modify highlight path to test data (X) --- dtreeviz/trees.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dtreeviz/trees.py b/dtreeviz/trees.py index 83d0908f..a38e4336 100644 --- a/dtreeviz/trees.py +++ b/dtreeviz/trees.py @@ -809,7 +809,7 @@ def instance_gr(): lcolor = colors['highlight'] lpw = "1.2" if node.right.id in highlight_path: - lcolor = colors['highlight'] + rcolor = colors['highlight'] rpw = "1.2" edges.append( f'{nname} -> {left_node_name} [penwidth={lpw} color="{lcolor}" label=<{llabel}>]' ) edges.append( f'{nname} -> {right_node_name} [penwidth={rpw} color="{rcolor}" label=<{rlabel}>]' )