Small fix to docstring in FlattenActionWrapper.
Small fix to docstring in FlattenActionWrapper.

PiperOrigin-RevId: 378875667
Change-Id: I0fda3ae1caa30fe57fdc6d6b55924202544be973

tensorflow/agents,tensorflow/agents