forked from ivadomed/ivadomed
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
58 lines (53 loc) · 2.2 KB
/
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
from setuptools import setup, find_packages
from codecs import open
from os import path
with open('requirements.txt') as f:
requirements = f.read().splitlines()
# Get README
this_directory = path.abspath(path.dirname(__file__))
with open(path.join(this_directory, 'README.md'), encoding='utf-8') as f:
long_description = f.read()
# Get Release version
path_version = path.join(this_directory, 'ivadomed', 'version.txt')
with open(path_version) as f:
version = f.read().strip()
setup(
name='ivadomed',
version=version,
description='Feature conditioning for IVADO medical imaging project.',
long_description=long_description,
long_description_content_type='text/markdown',
url='https://github.com/neuropoly/ivadomed',
author='NeuroPoly and Mila',
author_email='[email protected]',
classifiers=[
'Development Status :: 3 - Alpha',
'Intended Audience :: Developers',
'Programming Language :: Python :: 3',
],
python_requires='>=3.6, <3.9',
packages=find_packages(exclude=['docs', 'tests']),
include_package_data=True,
install_requires=requirements,
extras_require={
'docs': [ # pin sphinx to match what RTD uses:
# https://github.com/readthedocs/readthedocs.org/blob/ecac31de54bbb2c100f933e86eb22b0f4389ba84/requirements/pip.txt#L16
'sphinx<2',
'sphinx-rtd-theme<0.5',
],
'dev': ["pre-commit>=2.10.0"]
},
entry_points={
'console_scripts': [
'ivadomed=ivadomed.main:run_main',
'ivadomed_prepare_dataset_vertebral_labeling=ivadomed.scripts.prepare_dataset_vertebral_labeling:main',
'ivadomed_automate_training=ivadomed.scripts.automate_training:main',
'ivadomed_compare_models=ivadomed.scripts.compare_models:main',
'ivadomed_visualize_transforms=ivadomed.scripts.visualize_transforms:main',
'ivadomed_convert_to_onnx=ivadomed.scripts.convert_to_onnx:main',
'ivadomed_extract_small_dataset=ivadomed.scripts.extract_small_dataset:main',
'ivadomed_download_data=ivadomed.scripts.download_data:main',
'ivadomed_training_curve=ivadomed.scripts.training_curve:main'
],
},
)