Replace deprecated function in Python layer (#487)

dmlc · May 2, 2023 · 4e30fad · 4e30fad
1 parent af74664
commit 4e30fad
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/treelite/frontend.py b/python/treelite/frontend.py
@@ -539,8 +539,9 @@ def from_lightgbm(cls, booster):
         if not isinstance(booster, lightgbm.Booster):
             raise ValueError('booster must be of type `lightgbm.Booster`')
         model_str = booster.model_to_string()
-        _check_call(_LIB.TreeliteLoadLightGBMModelFromString(
+        _check_call(_LIB.TreeliteLoadLightGBMModelFromStringEx(
             c_str(model_str),
+            c_str("{}"),
             ctypes.byref(handle)))
         return Model(handle)