From e7e53306d0f1a778c40fac0f7df143ae86ef4688 Mon Sep 17 00:00:00 2001
From: Maxime Gasse <maxime.gasse@gmail.com>
Date: Wed, 23 Jun 2021 15:58:40 -0400
Subject: [PATCH] Allow MILP bipartite observations in the SOLVING stage

---
 libecole/src/observation/milpbipartite.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/libecole/src/observation/milpbipartite.cpp b/libecole/src/observation/milpbipartite.cpp
index ac74dc073..edfba2d56 100644
--- a/libecole/src/observation/milpbipartite.cpp
+++ b/libecole/src/observation/milpbipartite.cpp
@@ -134,7 +134,7 @@ template <typename T> auto vec_to_col(xt::xtensor<T, 1>&& t) -> xt::xtensor<T, 2
  *************************************/
 
 auto MilpBipartite::extract(scip::Model& model, bool /* done */) -> std::optional<MilpBipartiteObs> {
-	if (model.stage() < SCIP_STAGE_SOLVING) {
+	if (model.stage() <= SCIP_STAGE_SOLVING) {
 		auto [edge_features, constraint_features] = scip::get_all_constraints(model.get_scip_ptr(), normalize);
 
 		auto variable_features = xmatrix::from_shape({model.variables().size(), MilpBipartiteObs::n_variable_features});