Name 'base' is not defined

I opened the DQN tutorial in colab. And when I ran the following code in the “Input extraction” cell:

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


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

It’s a known issue tracked here.

Thanks. And this might be helpful.