Pass node_type to SRRegressor

MilesCranmer · Dec 19, 2023 · 72e5b02 · 72e5b02
1 parent 4f8eaeb
commit 72e5b02
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/CheckConstraints.jl b/src/CheckConstraints.jl
@@ -44,6 +44,7 @@ function count_max_nestedness(tree, degree, op)
         t -> (t.degree == degree && t.op == op) ? 1 : 0,  # Branches
         (p, c...) -> p + max(c...),  # Reduce
         tree,
+        break_sharing=Val(true),
     )
     # Remove count of self:
     is_self = tree.degree == degree && tree.op == op

diff --git a/src/MLJInterface.jl b/src/MLJInterface.jl
@@ -160,6 +160,7 @@ function _update(m, verbosity, old_fitresult, old_cache, X, y, w, options)
         y_t;
         niterations=m.niterations,
         weights=w_t,
+        node_type=m.node_type,
         variable_names=variable_names,
         options=options,
         parallelism=m.parallelism,