Method __init__

config/atari/env_wrapper.py:7–19

Atari Wrapper Parameters ---------- env: Any another env wrapper discount: float discount of env cvt_string: bool True -> convert the observation into string in the replay buffer

(self, env, discount: float, cvt_string=True)

class AtariWrapper(Game):
    def __init__(self, env, discount: float, cvt_string=True):
        """Atari Wrapper
        Parameters
        ----------
        env: Any
            another env wrapper
        discount: float
            discount of env
        cvt_string: bool
            True -> convert the observation into string in the replay buffer
        """
        super().__init__(env, env.action_space.n, discount)
        self.cvt_string = cvt_string

    def legal_actions(self):
        return [_ for _ in range(self.env.action_space.n)]