Skip to content

Commit

Permalink
cleand up old heatmap call
Browse files Browse the repository at this point in the history
  • Loading branch information
fernandopalafox committed Feb 5, 2024
1 parent df0bc57 commit 6a534ac
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions experiments/tower_defense.jl
Original file line number Diff line number Diff line change
Expand Up @@ -86,11 +86,11 @@ end
"""
Temp. script to calculate and plot heatmap of Stage 1 cost function
"""
function run_visualization(; type = "heatmap")
function run_visualization()
ps = [1 / 3, 1 / 3, 1 / 3]
βs = [[2, 1, 1], [1, 2, 1], [1, 1, 2]]
Ks = calculate_stage_1_costs(ps, βs)
fig = type == "heatmap" ? display_heatmap(ps, Ks) : display_surface(ps, Ks)
fig = display_surface(ps, Ks)
fig
end

Expand Down

0 comments on commit 6a534ac

Please sign in to comment.