diff --git a/flatland/envs/agent_chains.py b/flatland/envs/agent_chains.py index 52789f09f6c12437d34b03b3ed344f64ed7b485e..02734f463fd6566767226391e48772a30543bac4 100644 --- a/flatland/envs/agent_chains.py +++ b/flatland/envs/agent_chains.py @@ -182,14 +182,19 @@ class MotionCheck(object): def render(omc:MotionCheck, horizontal=True): - oAG = nx.drawing.nx_agraph.to_agraph(omc.G) - oAG.layout("dot") - sDot = oAG.to_string() - if horizontal: - sDot = sDot.replace('{', '{ rankdir="LR" ') - #return oAG.draw(format="png") - # This returns a graphviz object which implements __repr_svg - return gv.Source(sDot) + try: + oAG = nx.drawing.nx_agraph.to_agraph(omc.G) + oAG.layout("dot") + sDot = oAG.to_string() + if horizontal: + sDot = sDot.replace('{', '{ rankdir="LR" ') + #return oAG.draw(format="png") + # This returns a graphviz object which implements __repr_svg + return gv.Source(sDot) + except ImportError as oError: + print("Flatland agent_chains ignoring ImportError - install pygraphviz to render graphs") + return None + class ChainTestEnv(object): """ Just for testing agent chains