Merge pull request #62 from kento-forest/fix_highlight_path

fix : the right arrows in path to test data are not highlighted
parrt · Oct 20, 2019 · c95a160 · c95a160
2 parents 8aefb41 + 01cc13f
commit c95a160
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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}>]' )