Move count_nodes back to base.jl

Author: MilesCranmer

src/EquationUtils.jl

@@ -13,22 +13,6 @@ import ..EquationModule:
     any,
     filter_map
 
-"""
-    count_nodes(tree::AbstractNode)::Int
-
-Count the number of nodes in the tree.
-"""
-function count_nodes(tree::AbstractNode; break_sharing=Val(false))
-    return tree_mapreduce(
-        _ -> 1,
-        +,
-        tree,
-        Int64;
-        f_on_shared=(c, is_shared) -> is_shared ? 0 : c,
-        break_sharing,
-    )
-end
-
 """
     count_depth(tree::AbstractNode)::Int
 

src/base.jl

@@ -220,6 +220,22 @@ end
 # Derived functions: ##########################################################
 ###############################################################################
 
+"""
+    count_nodes(tree::AbstractNode)::Int
+
+Count the number of nodes in the tree.
+"""
+function count_nodes(tree::AbstractNode; break_sharing=Val(false))
+    return tree_mapreduce(
+        _ -> 1,
+        +,
+        tree,
+        Int64;
+        f_on_shared=(c, is_shared) -> is_shared ? 0 : c,
+        break_sharing,
+    )
+end
+
 """
     foreach(f::Function, tree::AbstractNode)