rename Trainer.on_message back to message to better match C++ code