modify highlight path to test data (X)

parrt · Oct 17, 2019 · 01cc13f · 01cc13f
1 parent e664301
commit 01cc13f
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}>]' )