--- a +++ b/extractiveSummarization/setup.py @@ -0,0 +1,61 @@ +from setuptools import setup, find_packages +import os +import sys +#TODO: nltk, spacy data dependency + +CURRENT_PYTHON = sys.version_info[:2] +REQUIRED_PYTHON = (3, 6) + +# This check and everything above must remain compatible with Python 2.7. +if CURRENT_PYTHON < REQUIRED_PYTHON: + sys.stderr.write(""" +========================== +Unsupported Python version +========================== +This version of Django requires Python {}.{}, but you're trying to +install it on Python {}.{}. +This may be because you are using a version of pip that doesn't +understand the python_requires classifier. Make sure you +have pip >= 9.0 and setuptools >= 24.2, then try again: + $ python -m pip install --upgrade pip setuptools + $ python -m pip install django +This will install the latest version of Django which works on your +version of Python. If you can't upgrade your pip (or Python), request +an older version of Django: + $ python -m pip install "django<2" +""".format(*(REQUIRED_PYTHON + CURRENT_PYTHON))) + sys.exit(1) + + + +with open('README.md') as f: + readme = f.read() + +with open('LICENSE') as f: + license = f.read() + +setup(name='ehrkit', + version='0.1.0', + description='NLP library for EHR data', + url='https://github.com/Yale-LILY/EHRKit', + author='LILY', + install_requires=[ + 'allennlp', + 'torch', + 'numpy', + 'sklearn', + 'pymysql', + 'sshtunnel', + 'nltk', + 'spacy', + 'gensim', + 'pandas', + 'bert-extractive-summarizer', + 'files2rouge', + 'scipy' + ], + # license='MIT', + packages=['ehrkit', 'external', 'summarizers', 'summarizers.evaluate'], + license=license, + long_description=readme +)