don't append threshold twice (#464)

Signed-off-by: Jan-Benedikt Jagusch <jan.jagusch@gmail.com>
onnx · Jun 18, 2021 · 582540e · 582540e
1 parent 680de9c
commit 582540e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/onnxmltools/convert/lightgbm/operator_converters/LightGbm.py b/onnxmltools/convert/lightgbm/operator_converters/LightGbm.py
@@ -96,7 +96,8 @@ def _parse_tree_structure(tree_id, class_id, learning_rate,
                 text = text[:100000] + "\n..."
             raise TypeError("threshold must be a number not '{}'"
                             "\n{}".format(tree_structure['threshold'], text))
-    attrs['nodes_values'].append(tree_structure['threshold'])
+    else:
+        attrs['nodes_values'].append(tree_structure['threshold'])
 
     # Assume left is the true branch and right is the false branch
     attrs['nodes_truenodeids'].append(left_id)