Skip to content

Commit 2dd65b3

Browse files
authored
Replace treelite.Model with treelite.frontend. (#2231)
1 parent b25dd40 commit 2dd65b3

File tree

1 file changed

+1
-1
lines changed
  • improver_tests/calibration/rainforests_calibration

1 file changed

+1
-1
lines changed

improver_tests/calibration/rainforests_calibration/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -280,7 +280,7 @@ def compile_models(lightgbm_models, lead_times, thresholds, tmp_path):
280280
for lead_time in lead_times:
281281
for threshold in thresholds:
282282
model = lightgbm_models[lead_time, threshold]
283-
treelite_model = treelite.Model.from_lightgbm(model)
283+
treelite_model = treelite.frontend.from_lightgbm(model)
284284
tl2cgen.export_lib(
285285
treelite_model,
286286
toolchain="gcc",

0 commit comments

Comments
 (0)