From a9119e513c389fde07eeb1f0817a0191c88112e9 Mon Sep 17 00:00:00 2001 From: dcradu <dcradu@uliege.be> Date: Mon, 19 Apr 2021 18:13:25 +0200 Subject: [PATCH] fixed bug on argparser for SGH --- src/jl/MCP_heuristics.jl | 6 ++++-- src/main.py | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/jl/MCP_heuristics.jl b/src/jl/MCP_heuristics.jl index be9d71b..c9c569c 100644 --- a/src/jl/MCP_heuristics.jl +++ b/src/jl/MCP_heuristics.jl @@ -372,7 +372,9 @@ function threshold_greedy_algorithm(D::Array{Float64,2}, c::Float64, n::Float64) obj_incumbent = obj_candidate locations_added += 1 end - x_incumbent = zeros(Float64, L) x_incumbent[ind_ones_incumbent] .= 1. return x_incumbent, obj_incumbent + x_incumbent = zeros(Float64, L) + x_incumbent[ind_incumbent] .= 1. + return x_incumbent, obj_incumbent end function time_threshold_greedy_algorithm(D::Array{Float64,2}, c::Float64, n::Float64) @@ -439,7 +441,7 @@ function stochastic_threshold_greedy_algorithm(D::Array{Float64,2}, c::Float64, locations_added += 1 end x_incumbent = zeros(Float64, L) - x_incumbent[ind_ones_incumbent] .= 1. + x_incumbent[ind_incumbent] .= 1. return x_incumbent, obj_incumbent end diff --git a/src/main.py b/src/main.py index 0413fa5..32d1f4e 100644 --- a/src/main.py +++ b/src/main.py @@ -17,7 +17,7 @@ def parse_args(): parser = argparse.ArgumentParser(description='Command line arguments.') parser.add_argument('--c', type=int) - parser.add_argument('--p', type=int, default=None) + parser.add_argument('--p', type=float, default=None) parser.add_argument('--run_BB', type=bool, default=False) parser.add_argument('--run_MIR', type=bool, default=False) parser.add_argument('--run_LS', type=bool, default=False) -- GitLab