diff --git a/gym/envs/toy_text/kellycoinflip.py b/gym/envs/toy_text/kellycoinflip.py index 172e6d148..aea0528ec 100644 --- a/gym/envs/toy_text/kellycoinflip.py +++ b/gym/envs/toy_text/kellycoinflip.py @@ -12,8 +12,8 @@ class KellyCoinflipEnv(gym.Env): self.action_space = spaces.Discrete(maxWealth*100) # betting in penny increments self.observation_space = spaces.Tuple(( - spaces.Discrete(maxWealth*100), # (w,b) - spaces.Discrete(maxRounds))) + spaces.Discrete(maxWealth*100+1), # (w,b) + spaces.Discrete(maxRounds+1))) self.reward_range = (0, maxWealth) self.edge = edge self.wealth = initialWealth