Add the imports to the example usage, so it's easier to copy and paste.
Add the imports to the example usage, so it's easier to copy and paste.

PiperOrigin-RevId: 277038516
Change-Id: Iaa627f7455540d55685f89d10eda86e209dddc5d

deepmind/dm_env