setup.py 4.09 KB
Newer Older
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
1
2
3
4
#!/usr/bin/env python
# vim: set fileencoding=utf-8 :
# Mon 16 Apr 08:18:08 2012 CEST

Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
5
6
from setuptools import setup, dist
dist.Distribution(dict(setup_requires=['bob.extension']))
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
7

Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
8
9
from bob.extension.utils import load_requirements, find_packages
install_requires = load_requirements()
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
10
11
12
13
14
15
16
17
18

# The only thing we do in this file is to call the setup() function with all
# parameters that define our package.
setup(

    # This is the basic information about your project. Modify all this
    # information before releasing code publicly.
    name = 'bob.learn.pytorch',
    version = open("version.txt").read().rstrip(),
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
19
20
    description = 'Bob bindings for PyTorch',
    url = 'https://gitlab.idiap.ch/bob/bob.learn.pytorch',
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
21
22
23
24
    license = 'BSD',
    author = 'Guillaume Heusch',
    author_email = 'guillaume.heusch@idiap.ch',
    keywords = 'pytorch',
25

Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
26
27
28
29
30
    # If you have a better, long description of your package, place it on the
    # 'doc' directory and then hook it here
    long_description = open('README.rst').read(),

    # This line is required for any distutils based packaging.
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
31
    packages = find_packages(),
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
32
    include_package_data = True,
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
33
    zip_safe=False,
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
34
35
36
37
38
39

    # This line defines which packages should be installed when you "install"
    # this package. All packages that are mentioned here, but are not installed
    # on the current system will be installed locally and only visible to the
    # scripts of this package. Don't worry - You won't need administrative
    # privileges when using buildout.
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
40
41
42
43
44
45
46
47
48
49
50
51
    install_requires = install_requires,

    # Your project should be called something like 'bob.<foo>' or
    # 'bob.<foo>.<bar>'. To implement this correctly and still get all your
    # packages to be imported w/o problems, you need to implement namespaces
    # on the various levels of the package and declare them here. See more
    # about this here:
    # http://peak.telecommunity.com/DevCenter/setuptools#namespace-packages
    #
    # Our database packages are good examples of namespace implementations
    # using several layers. You can check them out here:
    # https://www.idiap.ch/software/bob/packages
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
52

Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

    # This entry defines which scripts you will have inside the 'bin' directory
    # once you install the package (or run 'bin/buildout'). The order of each
    # entry under 'console_scripts' is like this:
    #   script-name-at-bin-directory = module.at.your.library:function
    #
    # The module.at.your.library is the python file within your library, using
    # the python syntax for directories (i.e., a '.' instead of '/' or '\').
    # This syntax also omits the '.py' extension of the filename. So, a file
    # installed under 'example/foo.py' that contains a function which
    # implements the 'main()' function of particular script you want to have
    # should be referred as 'example.foo:main'.
    #
    # In this simple example we will create a single program that will print
    # the version of bob.
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
68
69
70
    entry_points = {

      # scripts should be declared using this entry:
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
71
      'console_scripts' : [
72
        'train_cnn.py = bob.learn.pytorch.scripts.train_cnn:main',
73
        'train_mccnn.py = bob.learn.pytorch.scripts.train_mccnn:main',
Anjith GEORGE's avatar
Anjith GEORGE committed
74
        'train_generic.py = bob.learn.pytorch.scripts.train_generic:main',
75
        'train_fasnet.py = bob.learn.pytorch.scripts.train_fasnet:main',
76
77
        'train_dcgan.py = bob.learn.pytorch.scripts.train_dcgan:main',
        'train_conditionalgan.py = bob.learn.pytorch.scripts.train_conditionalgan:main',
78
        'train_network.py = bob.learn.pytorch.scripts.train_network:main',
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
79
80
81
82
83
84
85
      ],
   },

    # Classifiers are important if you plan to distribute this package through
    # PyPI. You can find the complete list of classifiers that are valid and
    # useful here (http://pypi.python.org/pypi?%3Aaction=list_classifiers).
    classifiers = [
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
86
      'Framework :: Bob',
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
87
88
      'Development Status :: 3 - Alpha',
      'Intended Audience :: Developers',
89
      'License :: OSI Approved :: GNU General Public License v3 (GPLv3)',
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
90
91
92
93
94
      'Natural Language :: English',
      'Programming Language :: Python',
      'Topic :: Scientific/Engineering :: Artificial Intelligence',
    ],
)