From d6caf8347844047d8b14118bf7429df82ed3900c Mon Sep 17 00:00:00 2001 From: Valentin Sulzer Date: Fri, 7 Feb 2020 13:41:20 -0500 Subject: [PATCH] #803 make visualise fail only raise logger error --- pybamm/expression_tree/symbol.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/pybamm/expression_tree/symbol.py b/pybamm/expression_tree/symbol.py index cd3ec77acf..3bc390cc26 100644 --- a/pybamm/expression_tree/symbol.py +++ b/pybamm/expression_tree/symbol.py @@ -285,9 +285,13 @@ def visualise(self, filename): new_node, counter = self.relabel_tree(self, 0) - DotExporter( - new_node, nodeattrfunc=lambda node: 'label="{}"'.format(node.label) - ).to_picture(filename) + try: + DotExporter( + new_node, nodeattrfunc=lambda node: 'label="{}"'.format(node.label) + ).to_picture(filename) + except FileNotFoundError: + # raise error but only through logger so that test passes + pybamm.logger.error("Please install graphviz>=2.42.2 to use dot exporter") def relabel_tree(self, symbol, counter): """ Finds all children of a symbol and assigns them a new id so that they can be