def __init__()

in env_utils.py [0:0]

• 10 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, env, k):
        """Stack k last frames.
        Returns lazy array, which is much more memory efficient."""
        gym.Wrapper.__init__(self, env)
        self.k = k
        self.frames = deque([], maxlen=k)
        shp = env.observation_space.shape
        self.observation_space = spaces.Box(
            low=0,
            high=255,
            shape=(shp[:-1] + (shp[-1] * k,)),
            dtype=env.observation_space.dtype)