Fixed weight being displayed on the edge of SplitBest nodes

cavalab · Aug 2, 2024 · 1bf8926 · 1bf8926
1 parent 8964234
commit 1bf8926
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/program/program.h b/src/program/program.h
@@ -419,7 +419,9 @@ template<PT PType> struct Program
                 // kid_id = kid_id.substr(2);
 
                 if (kid->data.get_is_weighted()
-                && Isnt<NodeType::Constant, NodeType::MeanLabel, NodeType::OffsetSum>(kid->data.node_type)){
+                && Isnt<NodeType::Constant, NodeType::MeanLabel, 
+                        NodeType::OffsetSum, NodeType::SplitBest>(kid->data.node_type))
+                {
                     edge_label = fmt::format("{:.2f}",kid->data.W);
                 }