Skip to content

Commit 0a38404

Browse files
dchetelatAntoinePrv
authored andcommitted
Changed SB to use vars rather than cols
1 parent 9b8ac20 commit 0a38404

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

libecole/src/observation/strongbranchingscores.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -68,12 +68,12 @@ std::optional<xt::xtensor<double, 1>> StrongBranchingScores::extract(scip::Model
6868
model.set_param("branching/vanillafullstrong/idempotent", idempotent);
6969

7070
/* Store strong branching scores in tensor */
71-
auto const num_lp_columns = static_cast<std::size_t>(SCIPgetNLPCols(scip));
72-
auto strong_branching_scores = xt::xtensor<double, 1>({num_lp_columns}, std::nan(""));
71+
auto const nb_vars = static_cast<std::size_t>(SCIPgetNVars(scip));
72+
auto strong_branching_scores = xt::xtensor<double, 1>({nb_vars}, std::nan(""));
7373

7474
for (auto const [var, score] : views::zip(cands, cands_scores)) {
75-
auto const lp_index = static_cast<std::size_t>(SCIPcolGetLPPos(SCIPvarGetCol(var)));
76-
strong_branching_scores[lp_index] = static_cast<double>(score);
75+
auto const var_index = static_cast<std::size_t>(SCIPvarGetProbindex(var));
76+
strong_branching_scores[var_index] = static_cast<double>(score);
7777
}
7878

7979
return strong_branching_scores;

0 commit comments

Comments
 (0)