Issue for CartPole

[ryanmaxwell96]

When I try to run the CartPole enviornment in pat-coady's code (pat-coady/trpo#32), I run into this error:

Traceback (most recent call last):
File "train.py", line 349, in
main(**vars(args))
File "train.py", line 289, in main
env, obs_dim, act_dim = init_gym(env_name)
File "train.py", line 72, in init_gym
act_dim = env.action_space.shape[0]
IndexError: tuple index out of range

After a little bit of digging I had to add self.action_space = np.array([1]) to line 53 in cartpole_bullet.py but I'm not sure if this is a correct fix to this problem. Any help?