saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-05-15 21:00:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update docstring

Browse source
This commit is contained in:
Antonin Raffin 2020-01-23 11:20:12 +01:00
parent 7265d9e352
commit fb57a6b80c
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
torchy_baselines/common/vec_env/vec_frame_stack.py
View file

@ -3,18 +3,18 @@ import warnings
import numpy as np
from gym import spaces
from torchy_baselines.common.vec_env.base_vec_env import VecEnvWrapper
from torchy_baselines.common.vec_env.base_vec_env import VecEnv, VecEnvWrapper
class VecFrameStack(VecEnvWrapper):
"""
Frame stacking wrapper for vectorized environment
:param venv: (VecEnv) the vectorized environment to wrap
:param n_stack: (int) Number of frames to stack
:param venv: the vectorized environment to wrap
:param n_stack: Number of frames to stack
"""
def __init__(self, venv, n_stack):
def __init__(self, venv: VecEnv, n_stack: int):
self.venv = venv
self.n_stack = n_stack
wrapped_obs_space = venv.observation_space