Skip to content

Commit cdd0237

Browse files
Clean up lower_varname
1 parent f2fd6f9 commit cdd0237

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

src/systems/diffeqs/first_order_transform.jl

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,12 @@
11
function lower_varname(var::Variable, naming_scheme; lower=false)
22
D = var.diff
3-
D == nothing && return var
3+
D === nothing && return var
44
order = lower ? D.order-1 : D.order
55
lower_varname(var.name, D.x, order, var.subtype, naming_scheme)
66
end
77
function lower_varname(sym::Symbol, idv, order::Int, subtype::Symbol, naming_scheme)
8-
order == 0 && return Variable(sym, subtype=subtype)
9-
name = Symbol(String(sym)*naming_scheme*String(idv.name)^order)
10-
Variable(name, subtype=subtype)
8+
name = order == 0 ? sym : Symbol(sym, naming_scheme, string(idv.name)^order)
9+
return Variable(name, subtype=subtype)
1110
end
1211

1312
function ode_order_lowering(sys::DiffEqSystem; kwargs...)

0 commit comments

Comments
 (0)