6af98bec16
Currently `pip install xgboost` will raise traceback like this
```
Traceback (most recent call last):
File "<string>", line 20, in <module>
File "/tmp/pip-build-IAdqYE/xgboost/setup.py", line 20, in <module>
import xgboost
File "./xgboost/__init__.py", line 8, in <module>
from .core import DMatrix, Booster
File "./xgboost/core.py", line 12, in <module>
import numpy as np
ImportError: No module named numpy
```
We should avoid importing numpy in setup.py and let pip install numpy and scipy automatically.
That's what `install_requires` for.
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
# pylint: disable=invalid-name
|
|
"""Setup xgboost package."""
|
|
from __future__ import absolute_import
|
|
import sys
|
|
from setuptools import setup, find_packages
|
|
import subprocess
|
|
sys.path.insert(0, '.')
|
|
|
|
import os
|
|
#build on the fly if install in pip
|
|
#otherwise, use build.sh in the parent directory
|
|
|
|
if 'pip' in __file__:
|
|
if not os.name == 'nt': #if not windows
|
|
build_sh = subprocess.Popen(['sh', 'xgboost/build-python.sh'])
|
|
build_sh.wait()
|
|
output = build_sh.communicate()
|
|
print(output)
|
|
|
|
|
|
CURRENT_DIR = os.path.dirname(__file__)
|
|
|
|
# We can not import `xgboost.libpath` in setup.py directly since xgboost/__init__.py
|
|
# import `xgboost.core` and finally will import `numpy` and `scipy` which are setup
|
|
# `install_requires`. That's why using `execfile` here.
|
|
libpath_py = os.path.join(CURRENT_DIR, 'xgboost/libpath.py')
|
|
libpath = {'__file__': libpath_py}
|
|
execfile(libpath_py, libpath, libpath)
|
|
|
|
LIB_PATH = libpath['find_lib_path']()
|
|
#print LIB_PATH
|
|
|
|
#to deploy to pip, please use
|
|
#make pythonpack
|
|
#python setup.py register sdist upload
|
|
#and be sure to test it firstly using "python setup.py register sdist upload -r pypitest"
|
|
setup(name='xgboost',
|
|
version=open(os.path.join(CURRENT_DIR, 'xgboost/VERSION')).read().strip(),
|
|
#version='0.4a13',
|
|
description=open(os.path.join(CURRENT_DIR, 'README.md')).read(),
|
|
install_requires=[
|
|
'numpy',
|
|
'scipy',
|
|
],
|
|
maintainer='Hongliang Liu',
|
|
maintainer_email='phunter.lau@gmail.com',
|
|
zip_safe=False,
|
|
packages=find_packages(),
|
|
#don't need this and don't use this, give everything to MANIFEST.in
|
|
#package_dir = {'':'xgboost'},
|
|
#package_data = {'': ['*.txt','*.md','*.sh'],
|
|
# }
|
|
#this will use MANIFEST.in during install where we specify additional files,
|
|
#this is the golden line
|
|
include_package_data=True,
|
|
data_files=[('xgboost', LIB_PATH)],
|
|
url='https://github.com/dmlc/xgboost')
|