Clarify error message of plot_model when pydot is not installed.
PiperOrigin-RevId: 350492746 Change-Id: Iad225857df0cd4ef46af70c68b3fef74ecc2783f
This commit is contained in:
parent
d1fe15249d
commit
514f050de0
@ -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.
|
||||
|
Loading…
Reference in New Issue
Block a user