Fixes for failing tests.

RaghuSpaceRajan · RaghuSpaceRajan · commit 6e398965f653 · 2021-06-04T20:07:35.000+02:00
diff --git a/setup.py b/setup.py
@@ -32,7 +32,9 @@
     'configspace==0.4.10',
     'scipy==1.3.0',
     'pandas==0.25.0',
-    'gym[atari]==0.14'
+    'gym[atari]==0.14',
+    'atari-py==0.2.5', # https://github.com/openai/atari-py/issues/81 #TODO Remove
+    'matplotlib', # #TODO Remove?
 ]
 
 extras_require_cont = [
