import torch
from torch import nn
from torch import optim
import torch.nn.functional as F
import matplotlib.pyplot as plt
from skimage.transform import resize
import numpy as np
from random import shuffle
from collections import deque
from IPython import display
import gym
#from nes_py.wrappers import BinarySpaceToDiscreteSpaceEnv
from nes_py.wrappers import JoypadSpace
import gym_super_mario_bros
from gym_super_mario_bros.actions import SIMPLE_MOVEMENT, COMPLEX_MOVEMENT
env = gym_super_mario_bros.make('SuperMarioBros-v0')
#env = BinarySpaceToDiscreteSpaceEnv(env, COMPLEX_MOVEMENT)
env = JoypadSpace(env, COMPLEX_MOVEMENT)
%matplotlib inline
--------------------------------------------------------------------------- ImportError Traceback (most recent call last) <ipython-input-6-cc57ffc7c90b> in <module>() 12 import gym 13 #from nes_py.wrappers import BinarySpaceToDiscreteSpaceEnv ---> 14 from nes_py.wrappers import JoypadSpace 15 import gym_super_mario_bros 16 from gym_super_mario_bros.actions import SIMPLE_MOVEMENT, COMPLEX_MOVEMENT ImportError: No module named wrappers
#Random agent. Just for testing
done = True
for step in range(5000):
if done:
state = env.reset()
state, reward, done, info = env.step(env.action_space.sample())
env.render()
#env.close()
--------------------------------------------------------------------------- NameError Traceback (most recent call last) <ipython-input-2-0c4d11147a88> in <module>() 3 for step in range(5000): 4 if done: ----> 5 state = env.reset() 6 state, reward, done, info = env.step(env.action_space.sample()) 7 env.render() NameError: name 'env' is not defined