forked from agi-brain/xuance
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
53 lines (52 loc) · 1.94 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
from setuptools import find_packages, setup
setup(
name='xuanpolicy',
packages=find_packages(include=['xuanpolicy', 'xuanpolicy.*']),
package_data={"xuanpolicy": ["configs/*.yaml", "configs/*/*/*.yaml"]},
version='0.1.7',
description='XuanPolicy: A Comprehensive Deep Reinforcement Learning Library.',
author='Wenzhang Liu, Wenzhe Cai, Kun Jiang, etc.',
author_email='',
license='MIT',
url='',
download_url='',
keywords=['deep reinforcement learning', 'software library', 'platform'],
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 >= 1.10.1"],
"all": [
"torch >= 1.13.0",
"tensorflow == 2.6.0",
"mindspore >= 1.10.1"
]
},
install_requires=[
"numpy >= 1.19.5",
"scipy >= 1.7.3",
"PyYAML >= 6.0",
"gym >= 0.21.0",
"gymnasium >= 0.28.1",
"mpi4py >= 3.1.3",
"tqdm >= 4.0",
"pyglet >= 1.5.15",
"opencv-python >= 4.5.4.58", # for Atari
"pettingzoo >= 1.23.0", # for MARL
"tensorboard >= 2.11.2" # logger
],
setup_requires=['pytest-runner'],
tests_requires=['pytest'],
test_suite='tests',
)