Fix rabit

2016-03-06 20:54:27 -08:00 · 2016-03-06 20:54:27 -08:00 · bec7332eea
commit bec7332eea
parent c05c5bc7bc
1 changed files with 4 additions and 3 deletions
--- a/python-package/xgboost/rabit.py
+++ b/python-package/xgboost/rabit.py
@ -9,6 +9,7 @@ from .core import _LIB, c_str, STRING_TYPES

 def _init_rabit():
    """internal libary initializer."""
+    if _LIB is not None:
        _LIB.RabitGetRank.restype = ctypes.c_int
        _LIB.RabitGetWorldSize.restype = ctypes.c_int
        _LIB.RabitVersionNumber.restype = ctypes.c_int