I try to run “REINFORCEMENT LEARNING (DQN) TUTORIAL” in Colab and get a NoSuchDisplayException: Cannot connect to “None”.

env.reset()
plt.figure()
plt.imshow(get_screen().cpu().squeeze(0).permute(1, 2, 0).numpy(),
interpolation=‘none’)
plt.title(‘Example extracted screen’)
plt.show()
NoSuchDisplayException                    Traceback (most recent call last)
 in ()
39 env.reset()
40 plt.figure()
—> 41 plt.imshow(get_screen().cpu().squeeze(0).permute(1, 2, 0).numpy(),
42            interpolation=‘none’)
43 plt.title(‘Example extracted screen’)
10 frames
/usr/local/lib/python3.6/dist-packages/pyglet/canvas/xlib.py in init(self, name, x_screen)
84         self._display = xlib.XOpenDisplay(name)
85         if not self._display:
—> 86             raise NoSuchDisplayException(‘Cannot connect to “%s”’ % name)
87
88         screen_count = xlib.XScreenCount(self._display)
NoSuchDisplayException: Cannot connect to “None”


I thought that the tutorials is adapted to the Colab, because there is even a link to launch them. What is an easy way to avoid this error? Or is it necessary to deal with pyvirtualdisplay?