Import Ray's changes to the python generator.