|
- 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.11',
- 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',
- )
|