Fix gym Atari ROMs

17_reinforcement_learning.ipynb

@@ -58,8 +58,9 @@
     "\n",
     "if IS_COLAB or IS_KAGGLE:\n",
     "    !apt update && apt install -y libpq-dev libsdl2-dev swig xorg-dev xvfb\n",
-    "    %pip install -q -U tf-agents pyvirtualdisplay gym[box2d,atari_py] AutoROM\n",
-    "    AUTOROM_PREFIX=\"/bin\n",
+    "    %pip install -U tf-agents pyvirtualdisplay\n",
+    "    %pip install -U gym>=0.21.0\n",
+    "    %pip install -U gym[box2d,atari,accept-rom-license]\n",
     "\n",
     "# Common imports\n",
     "import numpy as np\n",