setup.py 1001 Bytes
Newer Older
wxchan's avatar
wxchan committed
1
2
3
4
# coding: utf-8
# pylint: disable=invalid-name, exec-used
"""Setup lightgbm package."""
from __future__ import absolute_import
5

wxchan's avatar
wxchan committed
6
import os
7
8
9
import sys

from setuptools import find_packages, setup
10

wxchan's avatar
wxchan committed
11
12
13
14
15
16
17
18
sys.path.insert(0, '.')

CURRENT_DIR = os.path.dirname(__file__)

libpath_py = os.path.join(CURRENT_DIR, 'lightgbm/libpath.py')
libpath = {'__file__': libpath_py}
exec(compile(open(libpath_py, "rb").read(), libpath_py, 'exec'), libpath, libpath)

wxchan's avatar
wxchan committed
19
LIB_PATH = [os.path.relpath(path, CURRENT_DIR) for path in libpath['find_lib_path']()]
wxchan's avatar
wxchan committed
20
21
22
23
print("Install lib_lightgbm from: %s" % LIB_PATH)


setup(name='lightgbm',
24
      version=0.2,
wxchan's avatar
wxchan committed
25
26
27
28
29
30
31
32
33
34
35
36
      description="LightGBM Python Package",
      install_requires=[
          'numpy',
          'scipy',
      ],
      maintainer='Guolin Ke',
      maintainer_email='guolin.ke@microsoft.com',
      zip_safe=False,
      packages=find_packages(),
      include_package_data=True,
      data_files=[('lightgbm', LIB_PATH)],
      url='https://github.com/Microsoft/LightGBM')