Graphical models for Python