DQN pytorch official tutorial has an error on google colab

Question

i run pytorch official tutorial on google colab
screen = env.render(\'rgb_array\').transpose((2, 0, 1))
NameError: name \'base\' is not defined
.
based o