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

Merge pull request #930 from tqchen/master

Browse Source 
Fix rabit
This commit is contained in:
Tianqi Chen 2016-03-06 20:54:49 -08:00
commit 8c6cbe7608
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
python-package/xgboost/rabit.py
View File

@ -9,9 +9,10 @@ from .core import _LIB, c_str, STRING_TYPES
def _init_rabit():
"""internal libary initializer."""
_LIB.RabitGetRank.restype = ctypes.c_int
_LIB.RabitGetWorldSize.restype = ctypes.c_int
_LIB.RabitVersionNumber.restype = ctypes.c_int
if _LIB is not None:
_LIB.RabitGetRank.restype = ctypes.c_int
_LIB.RabitGetWorldSize.restype = ctypes.c_int
_LIB.RabitVersionNumber.restype = ctypes.c_int
def init(args=None):
"""Initialize the rabit libary with arguments"""