python: add train and pathos to model (matching erlang)