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,6 +1,6 @@
from collections import deque
import numpy as np
import warnings
from gym.spaces import Box
from gym import Wrapper
@@ -22,6 +22,7 @@ class LazyFrames(object):
__slots__ = ("frame_shape", "dtype", "shape", "lz4_compress", "_frames")
def __init__(self, frames, lz4_compress=False):
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.frame_shape = tuple(frames[0].shape)
self.shape = (len(frames),) + self.frame_shape
self.dtype = frames[0].dtype