BasicCollector: Make LP and MPS optional

miplearn/collectors/basic.py

@@ -16,6 +16,10 @@ from ..parallel import p_umap
 
 
 class BasicCollector:
+    def __init__(self, skip_lp: bool = False, write_mps: bool = True) -> None:
+        self.skip_lp = skip_lp
+        self.write_mps = write_mps
+
     def collect(
         self,
         filenames: List[str],
@@ -52,6 +56,7 @@ class BasicCollector:
                     model = build_model(data_filename)
                     model.extract_after_load(h5)
 
+                    if not self.skip_lp:
                         # Solve LP relaxation
                         relaxed = model.relax()
                         relaxed.optimize()
@@ -61,6 +66,7 @@ class BasicCollector:
                     model.optimize()
                     model.extract_after_mip(h5)
 
+                    if self.write_mps:
                         # Add lazy constraints to model
                         if model.lazy_enforce is not None:
                             model.lazy_enforce(model, model.lazy_)