From 0b8a5193956ce9dd624e8723d3fe908b2fd3527e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aur=C3=A9lien=20Geron?= <ageron@users.noreply.github.com>
Date: Mon, 18 Oct 2021 16:30:03 +1300
Subject: [PATCH] Fix gym Atari ROMs

---
 17_reinforcement_learning.ipynb | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/17_reinforcement_learning.ipynb b/17_reinforcement_learning.ipynb
index 334d1a3..12f185e 100644
--- a/17_reinforcement_learning.ipynb
+++ b/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",