-
Notifications
You must be signed in to change notification settings - Fork 2
/
setup.py
33 lines (28 loc) · 911 Bytes
/
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
import os.path
import sys
from setuptools import setup, find_packages
sys.path.append(os.path.join(os.path.dirname(__file__), 'reinforceflow'))
from reinforceflow.config import version
install_requires = [
'numpy',
'gym>=0.9.1',
'six',
'opencv-python'
]
extras_require = {
'tf': ['tensorflow>=1.6.0'],
'tf-gpu': ['tensorflow-gpu>=1.6.0'],
'universe': ['universe>=0.21.3']
}
setup(name='reinforceflow',
version=version,
description='Reinforcement Learning framework based on TensorFlow and OpenAI Gym',
url='https://github.com/dbobrenko/reinforceflow',
author='Dmytro Bobrenko',
author_email='[email protected]',
license='MIT',
packages=[package for package in find_packages()
if package.startswith('reinforceflow')],
install_requires=install_requires,
extras_require=extras_require,
zip_safe=False)