Visualization not working

[ethanabrooks]

When I run the script posted in your README:

from blackhc import mdp
from blackhc.mdp import example

spec = example.ONE_ROUND_DMDP

spec_graph = spec.to_graph()
spec_png = mdp.graph_to_png(spec_graph)

mdp.display_mdp(spec)

I get the following output:

<IPython.core.display.Image object>

No image is displayed and and no png file gets saved anywhere. Thank you for your help.