Skip to content

Commit 3d2e6ab

Browse files
vinitraWenbing Li
authored andcommitted
Error message should indicate lightgbm installation, not scikitlearn (#218)
1 parent c363106 commit 3d2e6ab

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onnxmltools/convert/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def convert_libsvm(model, name=None, initial_types=None, doc_string='', target_o
4343
def convert_lightgbm(model, name=None, initial_types=None, doc_string='', target_opset=None,
4444
targeted_onnx=onnx.__version__, custom_conversion_functions=None, custom_shape_calculators=None):
4545
if not utils.lightgbm_installed():
46-
raise RuntimeError('scikit-learn is not installed. Please install lightgbm to use this feature.')
46+
raise RuntimeError('lightgbm is not installed. Please install lightgbm to use this feature.')
4747

4848
from .lightgbm.convert import convert
4949
return convert(model, name, initial_types, doc_string, target_opset, targeted_onnx,

0 commit comments

Comments
 (0)