Updated setup.py to use setuptools

2020-01-27 22:23:04 -05:00 · 2020-01-27 22:23:04 -05:00 · 9b88b99841
commit 9b88b99841
parent 69503dafb1
1 changed files with 5 additions and 3 deletions
--- a/setup.py
+++ b/setup.py
@ -1,6 +1,8 @@
-from distutils.core import setup
+from setuptools import setup, find_packages
 setup(name='gymrecord',
      version='0.9',
-      scripts=['gymrecord'],
-      py_modules = ['play']
+      py_modules = ['play'],
+      scripts = ['gymrecord'],
+      packages=find_packages(),
+      install_requires = ['gym[atari]', 'pygame>=1,<2', 'matplotlib>=3,<4']
 )