Name 'base' is not defined

1

I opened the DQN tutorial in colab. And when I ran the following code in the “Input extraction” cell:
“”"
plt.figure()

plt.imshow(get_screen().cpu().squeeze(0).permute(1, 2, 0).numpy(),

       interpolation='none')

plt.title(‘Example extracted screen’)
“”".
I got this error. Anyone know how to fix it?

2

It’s a known issue tracked here.

3

Thanks. And this might be helpful.