CART export: use .left/.right instead of .child.left/.child.right

1 files changed, 2 insertions, 5 deletions

diff --git a/lib/functions.py b/lib/functions.py
index bfe0b00..de5c722 100644
--- a/lib/functions.py
+++ b/lib/functions.py
@@ -521,15 +521,12 @@ class CARTFunction(SKLearnRegressionFunction):
             ]
             sub_data["paramDecisionValue"] = tree.threshold[node_id]
             sub_data["type"] = "scalarSplit"
-            sub_data["child"] = {}
 
         # child value
         if left_child != self.leaf_id:
-            sub_data["child"]["left"] = self.recurse_(tree, left_child, depth=depth + 1)
+            sub_data["left"] = self.recurse_(tree, left_child, depth=depth + 1)
         if right_child != self.leaf_id:
-            sub_data["child"]["right"] = self.recurse_(
-                tree, right_child, depth=depth + 1
-            )
+            sub_data["right"] = self.recurse_(tree, right_child, depth=depth + 1)
 
         return sub_data