-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
68 lines (64 loc) · 2.55 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
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_custom'))
from version import __version__
# Environment-specific dependencies.
extras = {
# 'atari': ['atari_py~=0.2.0', 'Pillow', 'opencv-python'],
# 'box2d': ['box2d-py~=2.3.5'],
# 'classic_control': [],
'mujoco': ['mujoco_py>=1.50, <2.0', 'imageio'],
'robotics': ['mujoco_py>=1.50, <2.0', 'imageio'],
'custom': ['mujoco_py>=1.50, <2.0', 'imageio'],
'real': ['math3d', 'six'],
}
# Meta dependency groups.
extras['all'] = [item for group in extras.values() for item in group]
setup(name='gym-custom',
version=__version__,
description='Custom environments based on OpenAI Gym, a toolkit for developing and comparing your reinforcement learning agents.',
url='https://github.com/jigangkim/gym-custom',
author='Jigang Kim',
author_email='jgkim2020@snu.ac.kr',
license='',
packages=[package for package in find_packages()
if package.startswith('gym_custom')],
zip_safe=False,
install_requires=[
'scipy', 'numpy>=1.10.4', 'pyglet>=1.4.0,<=1.5.0', 'cloudpickle>=1.2.0,<1.4.0',
'enum34~=1.1.6;python_version<"3.4"', 'beepy', 'inputimeout'
],
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']
# },
package_data={'gym_custom': [
'envs/mujoco/assets/*.xml',
'envs/robotics/assets/fetch/*.xml',
'envs/robotics/assets/stls/fetch/*.stl',
'envs/robotics/assets/textures/*.png',
'env/custom/assets/*.xml',
'envs/custom/assets/ur3/*.xml',
'envs/custom/assets/meshes/objects/*.stl',
'envs/custom/assets/meshes/ur3/*.stl',
'envs/custom/assets/meshes/ur3/dual_ur3_stand_collision_box/*.stl',
'envs/custom/assets/textures/*.png']
},
tests_require=['pytest', 'mock'],
python_requires='>=3.5',
classifiers=[
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
]
)