-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
28 lines (26 loc) · 965 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
from setuptools import find_packages, setup
try: # for pip >= 10
from pip._internal.req import parse_requirements
except ImportError: # for pip <= 9.0.3
from pip.req import parse_requirements
import polus
install_reqs = parse_requirements("requirements.txt", session=False)
# https://stackoverflow.com/questions/62114945/attributeerror-parsedrequirement-object-has-no-attribute-req
#requirements = list(requirements)
try:
requirements = [str(ir.req) for ir in install_reqs]
except:
requirements = [str(ir.requirement) for ir in install_reqs]
setup(
name='polus',
packages=find_packages(include=['polus*']),
version=polus.__version__,
description='A tensorflow based framework for training deep learning models',
author='Tiago Almeida',
author_email='[email protected]',
license='MIT',
install_requires=requirements,
setup_requires=['pytest-runner'],
tests_require=['pytest'],
test_suite='tests',
)