[Feature Request] Add `group_map` support to MLAgents wrappers

[kurtamohler]

Motivation

Follow up from #2469, which did not add the group_map argument to the wrappers.

Checklist

• I have checked that there is no similar issue in the repo (required)