Clarify error message of plot_model when pydot is not installed.

PiperOrigin-RevId: 350492746
Change-Id: Iad225857df0cd4ef46af70c68b3fef74ecc2783f
This commit is contained in:
Francois Chollet 2021-01-06 22:35:01 -08:00 committed by TensorFlower Gardener
parent d1fe15249d
commit 514f050de0

View File

@ -104,9 +104,10 @@ def model_to_dot(model,
if not check_pydot():
message = (
'Failed to import pydot. You must `pip install pydot` '
'and install graphviz (https://graphviz.gitlab.io/download/), ',
'for `pydotprint` to work.')
'You must install pydot (`pip install pydot`) '
'and install graphviz '
'(see instructions at https://graphviz.gitlab.io/download/) ',
'for plot_model/model_to_dot to work.')
if 'IPython.core.magics.namespace' in sys.modules:
# We don't raise an exception here in order to avoid crashing notebook
# tests where graphviz is not available.