Skip to content

Commit 60d501b

Browse files
fix: use time_varying_as_func in build_function_wrapper
1 parent a845c2b commit 60d501b

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/systems/codegen_utils.jl

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,14 @@ function build_function_wrapper(sys::AbstractSystem, expr, args...; p_start = 2,
7474

7575
args = ntuple(Val(length(args))) do i
7676
arg = args[i]
77+
if is_time_dependent(sys)
78+
arg = if symbolic_type(arg) == NotSymbolic()
79+
arg isa AbstractArray ?
80+
map(x -> time_varying_as_func(unwrap(x), sys), arg) : arg
81+
else
82+
time_varying_as_func(unwrap(arg), sys)
83+
end
84+
end
7785
if symbolic_type(arg) == NotSymbolic() && arg isa AbstractArray
7886
DestructuredArgs(arg, generated_argument_name(i); create_bindings)
7987
else

0 commit comments

Comments
 (0)