Skip to content

Commit 4066428

Browse files
committed
fix: case of parameter_names as none
1 parent 99dfa91 commit 4066428

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

src/ParametricExpression.jl

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -71,8 +71,9 @@ function ParametricExpression(
7171
parameters::AbstractMatrix{T2},
7272
parameter_names,
7373
) where {T1,T2}
74-
@assert (isempty(parameters) && isnothing(parameter_names)) ||
75-
size(parameters, 1) == length(parameter_names)
74+
if !isnothing(parameter_names)
75+
@assert size(parameters, 1) == length(parameter_names)
76+
end
7677
T = promote_type(T1, T2)
7778
t = T === T1 ? tree : convert(ParametricNode{T}, tree)
7879
m = Metadata((;
@@ -283,10 +284,16 @@ function string_tree(
283284
UInt16(0)
284285
end
285286
end
287+
_parameter_names = ex.metadata.parameter_names
288+
parameter_names = if _parameter_names === nothing
289+
["p$(i)" for i in 1:num_params]
290+
else
291+
_parameter_names
292+
end
286293
variable_names3 = if variable_names2 === nothing
287-
vcat(["p$(i)" for i in 1:num_params], ["x$(i)" for i in 1:max_feature])
294+
vcat(parameter_names, ["x$(i)" for i in 1:max_feature])
288295
else
289-
vcat(ex.metadata.parameter_names, variable_names2)
296+
vcat(parameter_names, variable_names2)
290297
end
291298
@assert length(variable_names3) >= num_params + max_feature
292299
return string_tree(

0 commit comments

Comments
 (0)