Introduce set_default_variable_names!

Author: MilesCranmer

src/DynamicExpressions.jl

@@ -27,7 +27,7 @@ import Reexport: @reexport
     set_constants!
 @reexport import .OperatorEnumModule: AbstractOperatorEnum
 @reexport import .OperatorEnumConstructionModule:
-    OperatorEnum, GenericOperatorEnum, @extend_operators
+    OperatorEnum, GenericOperatorEnum, @extend_operators, set_default_variable_names!
 @reexport import .EvaluateEquationModule: eval_tree_array, differentiable_eval_tree_array
 @reexport import .EvaluateEquationDerivativeModule:
     eval_diff_tree_array, eval_grad_tree_array

src/OperatorEnumConstruction.jl

@@ -79,15 +79,19 @@ function _grad_evaluator(tree::Node, X; kws...)
     end
 end
 
+function set_default_variable_names!(variable_names::Vector{String})
+    return LATEST_VARIABLE_NAMES.x = variable_names
+end
+
 function create_evaluation_helpers!(operators::OperatorEnum)
     LATEST_OPERATORS.x = operators
     return LATEST_OPERATORS_TYPE.x = IsOperatorEnum
 end
-
 function create_evaluation_helpers!(operators::GenericOperatorEnum)
     LATEST_OPERATORS.x = operators
     return LATEST_OPERATORS_TYPE.x = IsGenericOperatorEnum
 end
+
 function lookup_op(@nospecialize(f), ::Val{degree}) where {degree}
     mapping = degree == 1 ? LATEST_UNARY_OPERATOR_MAPPING : LATEST_BINARY_OPERATOR_MAPPING
     if !haskey(mapping, f)

test/test_print.jl

@@ -111,7 +111,7 @@ end
     empty!(DynamicExpressions.OperatorEnumConstructionModule.LATEST_VARIABLE_NAMES.x)
     @test string(tree) == "((x1 * x2) + x3)"
     # Check if we can pass the wrong number of variable names:
-    DynamicExpressions.OperatorEnumConstructionModule.LATEST_VARIABLE_NAMES.x = ["k1"]
+    set_default_variable_names!(["k1"])
     @test string(tree) == "((k1 * x2) + x3)"
     empty!(DynamicExpressions.OperatorEnumConstructionModule.LATEST_VARIABLE_NAMES.x)
 end