Fixes according to feedback to pull #104

This commit is contained in:
Iaroslav Shcherbatyi
2016-05-28 11:30:40 +02:00
parent 7371debe38
commit 5c8e34029d
4 changed files with 52 additions and 24 deletions

View File

@@ -22,7 +22,8 @@ setup(name='gym',
'all': ['atari_py>=0.0.17', 'Pillow', 'pyglet', 'PyOpenGL',
'pachi-py>=0.0.19',
'box2d-py',
'mujoco_py>=0.4.3', 'imageio'],
'mujoco_py>=0.4.3', 'imageio',
'keras', 'theano'],
# Environment-specific dependencies. Keep these in sync with
# 'all'!
@@ -31,6 +32,7 @@ setup(name='gym',
'box2d': ['box2d-py'],
'classic_control': ['pyglet', 'PyOpenGL'],
'mujoco': ['mujoco_py>=0.4.3', 'imageio'],
'parameter_tuning': ['keras', 'theano'],
},
package_data={'gym': ['envs/mujoco/assets/*.xml', 'envs/classic_control/assets/*.png']},
tests_require=['nose2', 'mock'],