Farama-Foundation / SuperSuit

A collection of wrappers for Gymnasium and PettingZoo environments (being merged into gymnasium.wrappers and pettingzoo.wrappers
Other
441 stars 56 forks source link

Add full action masking support, expand and refactor testing #227

Closed elliottower closed 1 year ago

elliottower commented 1 year ago

See https://github.com/Farama-Foundation/SuperSuit/pull/221#issuecomment-1643247634 for more information

I added tests for all of the wrappers with AEC envs and they pretty much all work without modification actually, besides obs_delay which required a check for the obs to be a dict instead of np.ndarray.