cortex/setup.py at master · rdevon/cortex · GitHub

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 setup

packages = [
    'cortex', 'cortex._lib', 'cortex.built_ins', 'cortex._lib.data',
    'cortex.built_ins.datasets', 'cortex.built_ins.models',
    'cortex.built_ins.networks', 'cortex.built_ins.transforms']


install_requirements = [
    'imageio', 'matplotlib', 'progressbar2', 'scipy', 'sklearn', 'visdom',
    'pyyaml', 'pathlib', 'sphinxcontrib-napoleon', 'nibabel', 'torch', 'torchvision'
]

setup(name='cortex',
      version='0.11',
      description='A library for wrapping your pytorch code',
      author='R Devon Hjelm',
      author_email='erroneus@gmail.com',
      packages=packages,
      install_requires=[
          'imageio', 'matplotlib', 'progressbar2', 'scipy', 'sklearn',
          'torchvision', 'visdom', 'pyyaml', 'sphinxcontrib-napoleon'],
      entry_points={
          'console_scripts': [
              'cortex=cortex.main:run']
      },
      dependency_links={'git+https://github.com/facebookresearch/visdom.git'},
      zip_safe=False)