add opencv dependency for atari to fix tests

This commit is contained in:
Peter Zhokhov
2019-05-10 15:19:04 -07:00
parent 5861d95ecd
commit 05587f46d9

View File

@@ -7,7 +7,7 @@ from version import VERSION
# Environment-specific dependencies.
extras = {
'atari': ['atari_py>=0.1.4', 'Pillow', 'PyOpenGL'],
'atari': ['atari_py>=0.1.4', 'Pillow', 'PyOpenGL', 'opencv-python'],
'box2d': ['box2d-py>=2.3.5'],
'classic_control': ['PyOpenGL'],
'mujoco': ['mujoco_py>=1.50, <2.1', 'imageio'],