benchmark

0743d73f - Add pygame with gym requirement (#766)

Add pygame with gym requirement (#766) Summary: Currently the CI fails with "ModuleNotFoundError: No module named 'pygame'". The reason seems to be a recent update go gym, see https://github.com/openai/gym/issues/2634 Pull Request resolved: https://github.com/pytorch/benchmark/pull/766 Reviewed By: xuzhao9 Differential Revision: D34423163 Pulled By: kit1980 fbshipit-source-id: b4dc3ceb32c80d9e7d86c828441c6ddb28320e51