forked from rlworkgroup/garage
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
97 lines (90 loc) · 2.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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
from setuptools import find_packages
from setuptools import setup
# Required dependencies
required = [
# Please keep alphabetized
'akro==0.0.2',
'boto3',
'cached_property',
'click',
'cloudpickle',
'cma==1.1.06',
# dm_control throws an error during install about not being able to
# find a build dependency (absl-py). Later pip executes the `install`
# command again and the install succeeds because absl-py has been
# installed. This is stupid, but harmless.
'dm_control @ https://api.github.com/repos/deepmind/dm_control/tarball/7a36377879c57777e5d5b4da5aae2cd2a29b607a', # noqa: E501
'dowel==0.0.2',
'gym[all]==0.12.4',
'joblib<0.13,>=0.12',
'matplotlib',
'mujoco-py<2.1,>=2.0',
'numpy==1.14.5',
'psutil',
'pyprind',
'python-dateutil',
'torch==1.1.0',
'scikit-image',
'scipy',
'tensorflow<1.13,>=1.12.0',
'tensorflow-probability<0.6.0,>=0.5.0', # for tensorflow 1.12
'torchvision==0.3.0'
]
# Dependencies for optional features
extras = {}
extras['all'] = list(set(sum(extras.values(), [])))
# Intel dependencies not included in all
extras['intel'] = ['intel-tensorflow<1.13,>=1.12.0']
# Development dependencies (*not* included in "all")
extras['dev'] = [
# Please keep alphabetized
'baselines @ https://api.github.com/repos/openai/baselines/tarball/f2729693253c0ef4d4086231d36e0a4307ec1cb3', # noqa: E501
'coverage',
'flake8',
'flake8-docstrings==1.3.0',
'flake8-import-order',
'nose2',
'pandas',
'pep8-naming==0.7.0',
'pre-commit',
'pylint==1.9.2',
'pytest>=3.6', # Required for pytest-cov on Python 3.6
'pytest-cov',
'sphinx',
'sphinx_rtd_theme',
'yapf',
]
with open('README.md') as f:
readme = f.read()
# Get the package version dynamically
with open('VERSION') as v:
version = v.read().strip()
setup(
name='garage',
version=version,
author='Reinforcement Learning Working Group',
description='A framework for reproducible reinforcement learning research',
url='https://github.com/rlworkgroup/garage',
packages=find_packages(where='src'),
package_dir={'': 'src'},
scripts=['scripts/garage'],
install_requires=required,
extras_require=extras,
license='MIT',
long_description=readme,
long_description_content_type='text/markdown',
classifiers=[
'Development Status :: 4 - Beta',
'Intended Audience :: Developers',
'Intended Audience :: Education',
'Intended Audience :: Science/Research',
'License :: OSI Approved :: MIT License',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3 :: Only',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
'Topic :: Scientific/Engineering :: Mathematics',
'Topic :: Software Development :: Libraries',
],
)