Update rllib_models.py

[mhdmyz]

This change is required for creating create_env.py

[zhumingpassional]

have you tested the code? How about writing the unit test?

[mhdmyz]

All the codes I submitted have been tested and they work flawlessly together.