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

Fix SB3 compatibility wrapper (ProcConcatVec not returning info) #224

Closed elliottower closed 1 year ago

elliottower commented 1 year ago

This is to fix a minor bug somebody pointed out in our discord about different behavior for num_cpus=1 vs 2 or more, which was due to the ProcConcatVec env (multiprocessing) returning only obs, whereas the other ConcatVecEnv returned obs, info. This happened to work because SB3 expected only the observation with their 2.0.0 version, but the current bleeding edge version has fixed this problem and now correctly looks for both obs and info. So I am doing this to make a new release in preparation for finalizing our PettingZoo SB3 tutorials, and enhancing compatibility with SB3 going forward.