Fix all tests

miplearn/components/lazy_dynamic.py

@@ -85,8 +85,10 @@ class DynamicLazyConstraintsComponent(Component):
             disable=not sys.stdout.isatty(),
         ):
             logger.debug("Training: %s" % (str(v)))
-            label = np.zeros(len(training_instances))
-            label[violation_to_instance_idx[v]] = 1.0
+            label = [[True, False] for i in training_instances]
+            for idx in violation_to_instance_idx[v]:
+                label[idx] = [False, True]
+            label = np.array(label, dtype=np.bool8)
             classifier.fit(features, label)
 
     def predict(self, instance):

miplearn/components/steps/convert_tight.py

@@ -116,9 +116,11 @@ class ConvertTightIneqsIntoEqsStep(Component):
                 if category not in y:
                     y[category] = []
                 if 0 <= slack <= self.slack_tolerance:
-                    y[category] += [[1]]
+                    y[category] += [[False, True]]
                 else:
-                    y[category] += [[0]]
+                    y[category] += [[True, False]]
+            for category in y.keys():
+                y[category] = np.array(y[category], dtype=np.bool8)
         return y
 
     def predict(self, x):