Trainers: improve python bindings (#151)