openai / baselines

OpenAI Baselines: high-quality implementations of reinforcement learning algorithms

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[Question] Compatibilities wrappers for Atari games and gymnasium

TheGravityZero opened this issue · comments

Question

Where can I get these wrappers that are compatible with gymnasium?

from stable_baselines3.common.atari_wrappers import (
    ClipRewardEnv,
    EpisodicLifeEnv,
    FireResetEnv,
    MaxAndSkipEnv,
    NoopResetEnv,
)

If you use them from sb3, then there is a problem:

obs, reward, done,  info = self.env.step(action)
ValueError: too many values to unpack (expected 4)