ray/rllib/examples/partial_gpus.py