595a00466d
The setup.py is rewritten. This new script uses only Python code and provide customized implementation of setuptools commands. This way users can run most of setuptools commands just like any other Python libraries. * Remove setup_pip.py * Remove soft links. * Define customized commands. * Remove shell script. * Remove makefile script. * Update the doc for building from source.
61 lines
2.4 KiB
Python
61 lines
2.4 KiB
Python
# coding: utf-8
|
|
"""Find the path to xgboost dynamic library files."""
|
|
|
|
import os
|
|
import platform
|
|
import sys
|
|
|
|
|
|
class XGBoostLibraryNotFound(Exception):
|
|
"""Error thrown by when xgboost is not found"""
|
|
|
|
|
|
def find_lib_path():
|
|
"""Find the path to xgboost dynamic library files.
|
|
|
|
Returns
|
|
-------
|
|
lib_path: list(string)
|
|
List of all found library path to xgboost
|
|
"""
|
|
curr_path = os.path.dirname(os.path.abspath(os.path.expanduser(__file__)))
|
|
dll_path = [
|
|
# normal, after installation `lib` is copied into Python package tree.
|
|
os.path.join(curr_path, 'lib'),
|
|
# editable installation, no copying is performed.
|
|
os.path.join(curr_path, os.path.pardir, os.path.pardir, 'lib'),
|
|
]
|
|
if sys.platform == 'win32':
|
|
if platform.architecture()[0] == '64bit':
|
|
dll_path.append(
|
|
os.path.join(curr_path, '../../windows/x64/Release/'))
|
|
# hack for pip installation when copy all parent source
|
|
# directory here
|
|
dll_path.append(os.path.join(curr_path, './windows/x64/Release/'))
|
|
else:
|
|
dll_path.append(os.path.join(curr_path, '../../windows/Release/'))
|
|
# hack for pip installation when copy all parent source
|
|
# directory here
|
|
dll_path.append(os.path.join(curr_path, './windows/Release/'))
|
|
dll_path = [os.path.join(p, 'xgboost.dll') for p in dll_path]
|
|
elif sys.platform.startswith('linux') or sys.platform.startswith(
|
|
'freebsd'):
|
|
dll_path = [os.path.join(p, 'libxgboost.so') for p in dll_path]
|
|
elif sys.platform == 'darwin':
|
|
dll_path = [os.path.join(p, 'libxgboost.dylib') for p in dll_path]
|
|
elif sys.platform == 'cygwin':
|
|
dll_path = [os.path.join(p, 'cygxgboost.dll') for p in dll_path]
|
|
|
|
lib_path = [p for p in dll_path if os.path.exists(p) and os.path.isfile(p)]
|
|
|
|
# XGBOOST_BUILD_DOC is defined by sphinx conf.
|
|
if not lib_path and not os.environ.get('XGBOOST_BUILD_DOC', False):
|
|
link = 'https://xgboost.readthedocs.io/en/latest/build.html'
|
|
msg = 'Cannot find XGBoost Library in the candidate path. ' + \
|
|
'List of candidates:\n- ' + ('\n- '.join(dll_path)) + \
|
|
'\nXGBoost Python package path: ' + curr_path + \
|
|
'\nsys.prefix: ' + sys.prefix + \
|
|
'\nSee: ' + link + ' for installing XGBoost.'
|
|
raise XGBoostLibraryNotFound(msg)
|
|
return lib_path
|