Skip to content

Commit

Permalink
fix plotting
Browse files Browse the repository at this point in the history
  • Loading branch information
thorek1 committed Dec 22, 2023
1 parent e281680 commit 93acdde
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions src/plotting.jl
Original file line number Diff line number Diff line change
Expand Up @@ -1378,8 +1378,10 @@ function plot_conditional_forecast(𝓂::ℳ,

var_idx = indexin(var_names,full_SS)

var_names[indexin(𝓂.aux,var_names)] = map(x -> Symbol(replace(string(x), r"ᴸ⁽⁻?[⁰¹²³⁴⁵⁶⁷⁸⁹]+⁾" => "")), 𝓂.aux)

if length(intersect(𝓂.aux,var_names)) > 0
var_names[indexin(𝓂.aux,var_names)] = map(x -> Symbol(replace(string(x), r"ᴸ⁽⁻?[⁰¹²³⁴⁵⁶⁷⁸⁹]+⁾" => "")), 𝓂.aux)
end

relevant_SS = get_steady_state(𝓂, algorithm = algorithm, return_variables_only = true, derivatives = false)

relevant_SS = relevant_SS isa KeyedArray ? axiskeys(relevant_SS,1) isa Vector{String} ? rekey(relevant_SS, 1 => axiskeys(relevant_SS,1) .|> Meta.parse .|> replace_indices) : relevant_SS : relevant_SS
Expand Down

0 comments on commit 93acdde

Please sign in to comment.