Add pygame as a dependency since it is used by gym as of version 0.22.0
Add pygame as a dependency since it is used by gym as of version 0.22.0

PiperOrigin-RevId: 430158694
Change-Id: I9be574d2119f58e5f590580130c9a4eb949f9425

tensorflow/agents,tensorflow/agents