Try again

ankane · Oct 6, 2024 · eeb733e · eeb733e
1 parent df5fd62
commit eeb733e
Showing 1 changed file with 5 additions and 13 deletions.
diff --git a/ext/or-tools/constraint.cpp b/ext/or-tools/constraint.cpp
@@ -428,12 +428,11 @@ void init_constraint(Rice::Module& m) {
 
         std::atomic<bool> stopped(false);
 
-        if (!callback.is_nil()) {
-          // TODO figure out how to use callback with multiple cores
-          parameters.set_num_search_workers(1);
-
-          m.GetOrCreate<TimeLimit>()->RegisterExternalBooleanAsLimit(&stopped);
+        parameters.set_num_search_workers(1);
+        m.Add(NewSatParameters(parameters));
+        m.GetOrCreate<TimeLimit>()->RegisterExternalBooleanAsLimit(&stopped);
 
+        if (!callback.is_nil()) {
           m.Add(NewFeasibleSolutionObserver(
             [&](const CpSolverResponse& r) {
               // ensure Ruby thread
@@ -447,14 +446,7 @@ void init_constraint(Rice::Module& m) {
           );
         }
 
-        m.Add(NewSatParameters(parameters));
-        auto response = SolveCpModel(model.Build(), &m);
-
-        if (!callback.is_nil()) {
-          m.GetOrCreate<TimeLimit>()->RegisterExternalBooleanAsLimit(nullptr);
-        }
-
-        return response;
+        return SolveCpModel(model.Build(), &m);
       })
     .define_method(
       "_solution_integer_value",