fix #66

chsasank · chsasank · commit cefcbbb63b31 · 2017-04-14T22:41:43.000+05:30
diff --git a/intermediate_source/reinforcement_q_learning.py b/intermediate_source/reinforcement_q_learning.py
@@ -72,7 +72,7 @@
 import torch.nn.functional as F
 import torchvision.transforms as T
 
-env = gym.make('CartPole-v0')
+env = gym.make('CartPole-v0').unwrapped
 
 is_ipython = 'inline' in matplotlib.get_backend()
 if is_ipython:
