[python] Use predefined constant in feature importance type compariso…

…n instead of raw int literal (#5148)
microsoft · Apr 13, 2022 · 9a964fe · 9a964fe
1 parent 204fa9c
commit 9a964fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python-package/lightgbm/basic.py b/python-package/lightgbm/basic.py
@@ -3760,7 +3760,7 @@ def feature_importance(self, importance_type='split', iteration=None):
             ctypes.c_int(iteration),
             ctypes.c_int(importance_type_int),
             result.ctypes.data_as(ctypes.POINTER(ctypes.c_double))))
-        if importance_type_int == 0:
+        if importance_type_int == C_API_FEATURE_IMPORTANCE_SPLIT:
             return result.astype(np.int32)
         else:
             return result