Unverified Commit e35ed5f6 authored by AngelikaAntsmae's avatar AngelikaAntsmae Committed by GitHub
Browse files

[python-package] Improve Graphviz import error message (fixes #4299) (#4302)

parent c34b7ffd
......@@ -379,7 +379,7 @@ def _to_graphviz(tree_info, show_info, feature_names, precision=3,
if GRAPHVIZ_INSTALLED:
from graphviz import Digraph
else:
raise ImportError('You must install graphviz to plot tree.')
raise ImportError('You must install graphviz and restart your session to plot tree.')
def add(root, total_count, parent=None, decision=None):
"""Recursively add node or edge."""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment