mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-07-31 22:04:31 +00:00
59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
from setuptools import setup, find_packages
|
|
import sys, os.path
|
|
|
|
# Don't import gym module here, since deps may not be installed
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'gym'))
|
|
from version import VERSION
|
|
|
|
# Environment-specific dependencies.
|
|
extras = {
|
|
'atari': ['atari_py~=0.2.0', 'opencv-python>=3.'],
|
|
'box2d': ['box2d-py~=2.3.5'],
|
|
'classic_control': [],
|
|
'mujoco': ['mujoco_py>=1.50, <2.0', 'imageio'],
|
|
'robotics': ['mujoco_py>=1.50, <2.0', 'imageio'],
|
|
}
|
|
|
|
# Meta dependency groups.
|
|
extras['nomujoco'] = list(set([item for name, group in extras.items() if name != 'mujoco' and name != "robotics" for item in group]))
|
|
extras['all'] = list(set([item for group in extras.values() for item in group]))
|
|
|
|
setup(name='gym',
|
|
version=VERSION,
|
|
description='The OpenAI Gym: A toolkit for developing and comparing your reinforcement learning agents.',
|
|
url='https://github.com/openai/gym',
|
|
author='OpenAI',
|
|
author_email='gym@openai.com',
|
|
license='',
|
|
packages=[package for package in find_packages()
|
|
if package.startswith('gym')],
|
|
zip_safe=False,
|
|
install_requires=[
|
|
'scipy',
|
|
'numpy>=1.10.4',
|
|
'pyglet>=1.4.0',
|
|
'Pillow<=8.2.0',
|
|
'cloudpickle>=1.2.0,<1.7.0',
|
|
],
|
|
extras_require=extras,
|
|
package_data={'gym': [
|
|
'envs/mujoco/assets/*.xml',
|
|
'envs/classic_control/assets/*.png',
|
|
'envs/robotics/assets/LICENSE.md',
|
|
'envs/robotics/assets/fetch/*.xml',
|
|
'envs/robotics/assets/hand/*.xml',
|
|
'envs/robotics/assets/stls/fetch/*.stl',
|
|
'envs/robotics/assets/stls/hand/*.stl',
|
|
'envs/robotics/assets/textures/*.png']
|
|
},
|
|
tests_require=['pytest', 'mock'],
|
|
python_requires='>=3.6',
|
|
classifiers=[
|
|
'Programming Language :: Python :: 3',
|
|
'Programming Language :: Python :: 3.6',
|
|
'Programming Language :: Python :: 3.7',
|
|
'Programming Language :: Python :: 3.8',
|
|
'Programming Language :: Python :: 3.9',
|
|
],
|
|
)
|