setup.py 4.07 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',
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
89
      'License :: OSI Approved :: GPLv3 License',
Guillaume HEUSCH's avatar
Guillaume HEUSCH committed
90 91 92 93 94
      'Natural Language :: English',
      'Programming Language :: Python',
      'Topic :: Scientific/Engineering :: Artificial Intelligence',
    ],
)