Skip to content

Commit b6edca7

Browse files
fix: fix indexing related to array symbolics
1 parent 10bb71e commit b6edca7

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

src/parameter_indexing.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -628,7 +628,8 @@ function _getp(sys, ::ArraySymbolic, ::SymbolicTypeTrait, p)
628628
idx = parameter_index(sys, p)
629629
if is_timeseries_parameter(sys, p)
630630
ts_idx = timeseries_parameter_index(sys, p)
631-
return GetParameterTimeseriesIndex(idx, ts_idx)
631+
return GetParameterTimeseriesIndex(
632+
GetParameterIndex(idx), GetParameterIndex(ts_idx))
632633
else
633634
return GetParameterIndex(idx)
634635
end
@@ -750,5 +751,5 @@ function _setp_oop(indp, ::ArraySymbolic, ::SymbolicTypeTrait, sym)
750751
if is_parameter(indp, sym)
751752
return OOPSetter(_root_indp(indp), parameter_index(indp, sym), false)
752753
end
753-
error("$sym is not a valid parameter")
754+
return setp_oop(indp, collect(sym))
754755
end

src/state_indexing.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -468,7 +468,7 @@ function _setsym_oop(indp, ::ArraySymbolic, ::SymbolicTypeTrait, sym)
468468
return setsym_oop(indp, idx)
469469
elseif (idx = parameter_index(indp, sym)) !== nothing
470470
return FullSetter(
471-
nothing, OOPSetter(indp, idx isa AbstractArray ? idx : (idx,), false))
471+
nothing, OOPSetter(indp, idx, false))
472472
end
473473
return setsym_oop(indp, collect(sym))
474474
end

0 commit comments

Comments
 (0)