update setup.py

vwxyzjn · Apr 19, 2021 · 9817704 · 9817704
1 parent 6c9b371
commit 9817704
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,6 +1,7 @@
 *.tfevents.*
 wandb
 openaigym.*
+videos/*
 cleanrl/videos/*
 benchmark/**/*.svg
 benchmark/**/*.pkl

diff --git a/setup.py b/setup.py
@@ -2,10 +2,14 @@
 import versioneer
 
 extras = {
-  'gym_all': ['gym[all]', 'opencv-python'],
+  'gym_all': ['gym[atari]', 'gym[box2d]', 'opencv-python'],
+  'atari': ['gym[atari]'],
+  'box2d': ['gym[box2d]', 'opencv-python'],
   'pybullet': ['pybullet'],
   'pysc2': ['pysc2'],
-  'procgen': ['procgen']
+  'procgen': ['procgen'],
+  'cloud': ['boto3'],
+  'gym_microrts': ['gym_microrts'],
 }
 
 extras['all'] = [item for group in extras.values() for item in group]