您最多选择25个主题
主题必须以中文或者字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
52 行
1.5 KiB
52 行
1.5 KiB
from setuptools import setup, find_packages
|
|
from os import path
|
|
from io import open
|
|
|
|
here = path.abspath(path.dirname(__file__))
|
|
|
|
# Get the long description from the README file
|
|
with open(path.join(here, 'README.md'), encoding='utf-8') as f:
|
|
long_description = f.read()
|
|
|
|
setup(
|
|
name='mlagents',
|
|
version='0.8.1',
|
|
description='Unity Machine Learning Agents',
|
|
long_description=long_description,
|
|
long_description_content_type='text/markdown',
|
|
url='https://github.com/Unity-Technologies/ml-agents',
|
|
author='Unity Technologies',
|
|
author_email='ML-Agents@unity3d.com',
|
|
|
|
classifiers=[
|
|
'Intended Audience :: Developers',
|
|
'Topic :: Scientific/Engineering :: Artificial Intelligence',
|
|
'License :: OSI Approved :: Apache Software License',
|
|
'Programming Language :: Python :: 3.6'
|
|
],
|
|
|
|
packages=['mlagents.trainers', 'mlagents.trainers.bc', 'mlagents.trainers.ppo'], # Required
|
|
zip_safe=False,
|
|
|
|
install_requires=[
|
|
'mlagents_envs==0.8.1',
|
|
'tensorflow>=1.7,<1.8',
|
|
'Pillow>=4.2.1',
|
|
'matplotlib',
|
|
'numpy>=1.13.3,<=1.14.5',
|
|
'jupyter',
|
|
'pytest>=3.2.2,<4.0.0',
|
|
'docopt',
|
|
'pyyaml',
|
|
'protobuf>=3.6,<3.7',
|
|
'grpcio>=1.11.0,<1.12.0',
|
|
'pypiwin32==223;platform_system=="Windows"'],
|
|
|
|
python_requires=">=3.6,<3.7",
|
|
|
|
entry_points={
|
|
'console_scripts': [
|
|
'mlagents-learn=mlagents.trainers.learn:main',
|
|
],
|
|
},
|
|
)
|