RoBO/setup.py at master · saralajew/RoBO · 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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
import os
import sys
from setuptools import setup, find_packages
here = os.path.abspath(os.path.dirname(__file__))

requires = [
    'DIRECT',
    'george',
    'emcee',
    'cma',
    'scipy >= 0.12',
    'numpy >= 1.7'
    ]
dependency_links = [
      'https://github.com/sfalkner/george/tarball/master#egg=george'
      ]

optional_dependencies = [
    'cython',
    'rf',
    'matplotlib',
    'theano',
    'lasagne',
    'sgmcmc',
    'pymatbridge'
    ]

opt_dependency_links = {
    'rf': 'https://bitbucket.org/aadfreiburg/random_forest_run/get/master.zip#egg=rf'
}

user_provided_depend = sys.argv[2:len(sys.argv)]
for i in user_provided_depend:
    dependency_name = i[2:len(i)]
    if dependency_name in optional_dependencies:
        if dependency_name in opt_dependency_links:
            dependency_links.append(opt_dependency_links[dependency_name])
        requires.append(dependency_name)
        sys.argv.remove(i)

setup(name='RoBO',
      version='0.2.1',
      description='Framework for Bayesian optimization',
      author='Aaron Klein',
      author_email='[email protected]',
      url='http://automl.github.io/RoBO/',
      keywords='Bayesian Optimization',
      packages=find_packages(),
      license='LICENSE.txt',
      test_suite='robo',
      install_requires=requires,
      dependency_links=dependency_links)