-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
30 lines (27 loc) · 926 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
import sys, os
from setuptools import setup, find_packages
'''
Symmetric Play Setup Script
Notes for later: see package_data arg if additional files need to be supplied.
'''
if sys.version_info.major != 3:
print('Please use Python3!')
setup(name='symmetric_play',
packages=[package for package in find_packages()
if package.startswith('symmetric_play')],
install_requires=[
'gym[atari,classic_control]==0.15.3',
'stable_baselines[mpi]==2.9.0',
'numpy<1.17',
'matplotlib',
'seaborn'],
extras_require={
'cpu' : ['tensorflow==1.14.0'],
'gpu' : ['tensorflow-gpu==1.14.0'],
},
description='Framework for Symmetric Play Experiments',
author='Joey Hejna',
url='https://github.com/jhejna/symmetric_play',
lisence='MIT',
version='0.0.1',
)