Fix issue #1

vidalt · Mar 7, 2024 · 19273ec · 19273ec
1 parent 886be50
commit 19273ec
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/DRAFT.py b/DRAFT.py
@@ -351,7 +351,8 @@ def perform_reconstruction_v1_CP_SAT(self, n_threads=0, time_out=60, verbosity=1
                 for c in range(C):
                     model.Add(cp_model.LinearExpr.Sum(branch_vars_c[c]) ==  int(a_branch[1][c])) # enforces the branch per-class cardinality
             for k in range(N):
-                model.Add(cp_model.LinearExpr.Sum(examples_capts[k]) == 1) # each example captured by exactly one branch
+                if len(examples_capts[k]) > 0:
+                    model.Add(cp_model.LinearExpr.Sum(examples_capts[k]) == 1) # each example captured by exactly one branch
 
         # Ordre lexicographique au sein des classes
         '''deb = 0