in understanding_rl_vision/rl_clarity/training.py [0:0]
def __init__(self, venv, *, extra_actions, default_action):
assert isinstance(venv.action_space, gym.spaces.Discrete)
super().__init__(
venv, action_space=gym.spaces.Discrete(venv.action_space.n + extra_actions)
)
self.default_action = default_action