/
setup.py
95 lines (89 loc) · 2.68 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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
# Copyright 2019 IBM Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from setuptools import setup, find_packages
import os
import sys
import logging
logger = logging.getLogger(__name__)
logger.addHandler(logging.StreamHandler())
try:
import builtins
# This trick is borrowed from scikit-learn
# This is a bit (!) hackish: we are setting a global variable so that the
# main lale __init__ can detect if it is being loaded by the setup
# routine, to avoid attempting to import components before installation.
builtins.__LALE_SETUP__ = True
except ImportError:
pass
with open("README.md", "r") as fh:
long_description = fh.read()
on_rtd = os.environ.get('READTHEDOCS') == 'True'
if on_rtd:
install_requires = []
else:
install_requires=[
'astunparse',
'graphviz',
'hyperopt==0.2.3',
'jsonschema',
'jsonsubschema',
'numpy',
'scikit-learn==0.20.3',
'scipy',
'pandas<=0.25.3',
'xgboost',
'lightgbm',
'decorator',
'h5py']
import lale
VERSION=lale.__version__
setup(
name='lale',
version=VERSION,
author="Guillaume Baudart, Martin Hirzel, Kiran Kate, Parikshit Ram, Avraham Shinnar",
description="Library for Semi-Automated Data Science",
long_description=long_description,
long_description_content_type="text/markdown",
url="https://github.com/IBM/lale",
python_requires='>=3.6',
packages=find_packages(),
license='',
install_requires = install_requires,
extras_require={
'full': [
'liac-arff>=2.4.0',
'pytorch-pretrained-bert>=0.6.1',
'torchvision>=0.2.2',
'tensorflow-datasets>=1.0.1',
'tensorflow>=1.13.1',
'tensorflow_hub',
'spacy',
'smac<=0.10.0',
'aif360',
'numba',
'torch>=1.0',
'BlackBoxAuditing',
'imbalanced-learn==0.4.3'],
'test':[
'autoai-libs',
'jupyter',
'mypy',
'flake8',
'numpydoc',
'sphinx',
'm2r',
'sphinx_rtd_theme',
'sphinxcontrib.apidoc'
]}
)