@@ -142,10 +142,15 @@ end
142142get_contents (ex:: ParametricExpression ) = ex. tree
143143get_metadata (ex:: ParametricExpression ) = ex. metadata
144144get_tree (ex:: ParametricExpression ) = ex. tree
145- function get_operators (ex:: ParametricExpression , operators= nothing )
145+ function get_operators (
146+ ex:: ParametricExpression , operators:: Union{AbstractOperatorEnum,Nothing} = nothing
147+ )
146148 return operators === nothing ? ex. metadata. operators : operators
147149end
148- function get_variable_names (ex:: ParametricExpression , variable_names= nothing )
150+ function get_variable_names (
151+ ex:: ParametricExpression ,
152+ variable_names:: Union{Nothing,AbstractVector{<:AbstractString}} = nothing ,
153+ )
149154 return variable_names === nothing ? ex. metadata. variable_names : variable_names
150155end
151156@inline _copy_with_nothing (x) = copy (x)
@@ -241,7 +246,10 @@ function eval_tree_array(::ParametricExpression{T}, ::AbstractMatrix{T}, operato
241246end
242247# ! format: on
243248function (ex:: ParametricExpression )(
244- X:: AbstractMatrix{T} , classes:: AbstractVector{<:Integer} , operators= nothing ; kws...
249+ X:: AbstractMatrix{T} ,
250+ classes:: AbstractVector{<:Integer} ,
251+ operators:: Union{AbstractOperatorEnum,Nothing} = nothing ;
252+ kws... ,
245253) where {T}
246254 (output, flag) = eval_tree_array (ex, X, classes, operators; kws... ) # Will error
247255 if ! flag
@@ -253,7 +261,7 @@ function eval_tree_array(
253261 ex:: ParametricExpression{T} ,
254262 X:: AbstractMatrix{T} ,
255263 classes:: AbstractVector{<:Integer} ,
256- operators= nothing ;
264+ operators:: Union{AbstractOperatorEnum,Nothing} = nothing ;
257265 kws... ,
258266) where {T}
259267 @assert length (classes) == size (X, 2 )
0 commit comments