"src/vscode:/vscode.git/clone" did not exist on "ddcbe71cddf7568e8f66884ec2f74242c51db88d"
libpath.py 1.46 KB
Newer Older
wxchan's avatar
wxchan committed
1
2
3
4
# coding: utf-8
"""Find the path to lightgbm dynamic library files."""
import os

5
6
from platform import system

wxchan's avatar
wxchan committed
7
8
9
10
11
12
13
14

def find_lib_path():
    """Find the path to LightGBM library files.
    Returns
    -------
    lib_path: list(string)
       List of all found library path to LightGBM
    """
15
16
17
18
    if os.environ.get('LIGHTGBM_BUILD_DOC', False):
        # we don't need lib_lightgbm while building docs
        return []

wxchan's avatar
wxchan committed
19
    curr_path = os.path.dirname(os.path.abspath(os.path.expanduser(__file__)))
20
21
22
23
    dll_path = [curr_path, os.path.join(curr_path, '../../'),
                os.path.join(curr_path, 'compile'),
                os.path.join(curr_path, '../compile'),
                os.path.join(curr_path, '../../lib/')]
24
    if system() in ('Windows', 'Microsoft'):
25
26
        dll_path.append(os.path.join(curr_path, '../compile/Release/'))
        dll_path.append(os.path.join(curr_path, '../compile/windows/x64/DLL/'))
Guolin Ke's avatar
Guolin Ke committed
27
        dll_path.append(os.path.join(curr_path, '../../Release/'))
28
        dll_path.append(os.path.join(curr_path, '../../windows/x64/DLL/'))
wxchan's avatar
wxchan committed
29
30
31
32
33
        dll_path = [os.path.join(p, 'lib_lightgbm.dll') for p in dll_path]
    else:
        dll_path = [os.path.join(p, 'lib_lightgbm.so') for p in dll_path]
    lib_path = [p for p in dll_path if os.path.exists(p) and os.path.isfile(p)]
    if not lib_path:
34
        dll_path = [os.path.realpath(p) for p in dll_path]
35
        raise Exception('Cannot find lightgbm library in following paths: ' + '\n'.join(dll_path))
wxchan's avatar
wxchan committed
36
    return lib_path