deprecate built in wrappers for supersuit

This commit is contained in:
Justin Terry
2021-07-28 22:21:47 -04:00
parent bc76034167
commit 3133e99a64
15 changed files with 35 additions and 12 deletions

View File

@@ -1,5 +1,5 @@
import numpy as np
import warnings
from gym.spaces import Box
from gym import ObservationWrapper
@@ -12,6 +12,7 @@ class ResizeObservation(ObservationWrapper):
if isinstance(shape, int):
shape = (shape, shape)
assert all(x > 0 for x in shape), shape
warnings.warn("Gym\'s internal preprocessing wrappers are now deprecated. While they will continue to work for the foreseeable future, we strongly recommend using SuperSuit instead: https://github.com/PettingZoo-Team/SuperSuit")
self.shape = tuple(shape)
obs_shape = self.shape + self.observation_space.shape[2:]