forked from agi-brain/xuance
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
71 lines (70 loc) · 2.65 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
69
70
71
from setuptools import find_packages, setup
setup(
name='xuance',
packages=find_packages(include=['xuance', 'xuance.*']),
package_data={
"xuance":
[
"configs/*.yaml",
"configs/*/*.yaml",
"configs/*/*/*.yaml",
"environment/magent2/libmagent.so", # for magent2 environment on linux
"environment/magent2/magent.dll", # for magent2 environment on Windows
"environment/magent2/libmagent.dylib" # for magent2 environment on MacOS
]
},
version="1.0.10",
description='XuanCe: A Comprehensive and Unified Deep Reinforcement Learning Library.',
author='XuanCe contributors.',
author_email='',
license='MIT',
url='',
download_url='https://github.com/agi-brain/xuance.git',
keywords=['deep reinforcement learning', 'software library', 'PyTorch', 'TensorFlow2', 'MindSpore'],
classifiers=[
'Development Status :: 4 - Beta',
# Chose either "3 - Alpha", "4 - Beta" or "5 - Production/Stable" as the current state of your package
'Intended Audience :: Developers', # Define that your audience are developers
'Topic :: Software Development :: Build Tools',
'License :: OSI Approved :: MIT License', # Again, pick a license
'Programming Language :: Python :: 3.6', # Specify which pyhton versions that you want to support
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
'Programming Language :: Python :: 3.9',
'Programming Language :: Python :: 3.10',
],
extras_require={
"torch": ["torch==1.13.0"],
"tensorflow": ["tensorflow==2.6.0"],
"mindspore": ["mindspore==2.2.0"],
"all": [
"torch==1.13.0",
"tensorflow==2.6.0",
"mindspore==2.2.0" # mindspore might be installed manually.
]
},
install_requires=[
"numpy>=1.21.6",
"scipy==1.7.3",
"PyYAML", # default version is 6.0
"gym==0.26.2",
"gymnasium==0.28.1",
"gym-notices==0.0.8",
# "box2d-py==2.3.5", # for box2d
"mpi4py", # default version is 3.1.3
"tqdm==4.62.3",
"pyglet==1.5.15",
"opencv-python==4.5.4.58", # for Atari
"atari-py==0.2.9",
"ale-py==0.7.5",
"pettingzoo>=1.23.0", # for MARL
# "magent2", # 0.3.2 is suggested
"tensorboard==2.11.2", # logger
"wandb==0.15.3",
"moviepy==1.0.3",
"imageio" # default version is 2.9.0
],
setup_requires=['pytest-runner'],
tests_requires=['pytest'],
test_suite='tests',
)