Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Better error message for failed library loading (#3690)

Browse Source 
* Better error message for failed lib loading

* Address review comment + fix lint
This commit is contained in:
Philip Hyunsu Cho 2018-09-12 22:37:26 -07:00 committed by GitHub
parent 3209b42b07
commit bd41bd6605
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
python-package/xgboost/core.py
View File

@ -118,13 +118,27 @@ def _load_lib():
if len(lib_paths) == 0: if len(lib_paths) == 0:
return None return None
pathBackup = os.environ['PATH'] pathBackup = os.environ['PATH']
lib_success = False
os_error_list = []
for lib_path in lib_paths: for lib_path in lib_paths:
try: try:
# needed when the lib is linked with non-system-available dependencies # needed when the lib is linked with non-system-available dependencies
os.environ['PATH'] = pathBackup + os.pathsep + os.path.dirname(lib_path) os.environ['PATH'] = pathBackup + os.pathsep + os.path.dirname(lib_path)
lib = ctypes.cdll.LoadLibrary(lib_path) lib = ctypes.cdll.LoadLibrary(lib_path)
except OSError: lib_success = True
except OSError as e:
os_error_list.append(str(e))
continue continue
if not lib_success:
libname = os.path.basename(lib_paths[0])
raise XGBoostError(
'XGBoost Library ({}) could not be loaded.\n'.format(libname) +
'Likely causes:\n' +
' * OpenMP runtime is not installed ' +
'(vcomp140.dll or libgomp-1.dll for Windows, ' +
'libgomp.so for UNIX-like OSes)\n' +
' * You are running 32-bit Python on a 64-bit OS\n' +
'Error message(s): {}\n'.format(os_error_list))
lib.XGBGetLastError.restype = ctypes.c_char_p lib.XGBGetLastError.restype = ctypes.c_char_p
lib.callback = _get_log_callback_func() lib.callback = _get_log_callback_func()
if lib.XGBRegisterLogCallback(lib.callback) != 0: if lib.XGBRegisterLogCallback(lib.callback) != 0: