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

Fix rabit

Browse Source
This commit is contained in:
tqchen 2016-03-06 20:54:27 -08:00
parent c05c5bc7bc
commit bec7332eea
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(): def _init_rabit():
"""internal libary initializer.""" """internal libary initializer."""
_LIB.RabitGetRank.restype = ctypes.c_int if _LIB is not None:
_LIB.RabitGetWorldSize.restype = ctypes.c_int _LIB.RabitGetRank.restype = ctypes.c_int
_LIB.RabitVersionNumber.restype = ctypes.c_int _LIB.RabitGetWorldSize.restype = ctypes.c_int
_LIB.RabitVersionNumber.restype = ctypes.c_int
def init(args=None): def init(args=None):
"""Initialize the rabit libary with arguments""" """Initialize the rabit libary with arguments"""