Support IBM i OS (#7920)

This commit is contained in:
Gavin Zhang
2022-06-02 23:38:35 +08:00
committed by GitHub
parent 31e6902e43
commit 6426449c8b
3 changed files with 9 additions and 0 deletions

View File

@@ -44,6 +44,8 @@ def lib_name() -> str:
name = 'libxgboost.dylib'
elif system() == 'Windows':
name = 'xgboost.dll'
elif system() == 'OS400':
name = 'libxgboost.so'
return name

View File

@@ -49,6 +49,8 @@ def find_lib_path() -> List[str]:
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]
if platform.system() == 'OS400':
dll_path = [os.path.join(p, 'libxgboost.so') for p in dll_path]
lib_path = [p for p in dll_path if os.path.exists(p) and os.path.isfile(p)]