Example of using a custom gym env with RLlib.