Fix failing tests

src/python/miplearn/benchmark.py

@@ -28,8 +28,7 @@ class BenchmarkRunner:
         for (name, solver) in self.solvers.items():
             results = solver.parallel_solve(instances,
                                             n_jobs=n_jobs,
-                                            label="Solve (%s)" % name,
+                                            label="Solve (%s)" % name)
-                                            collect_training_data=False)
             for i in range(len(instances)):
                 self._push_result(results[i],
                                   solver=solver,

src/python/miplearn/tests/test_solver.py

@@ -2,6 +2,9 @@
 #  Copyright (C) 2020, UChicago Argonne, LLC. All rights reserved.
 #  Released under the modified BSD license. See COPYING.md for more details.
 
+import pickle
+import tempfile
+
 from miplearn import LearningSolver, BranchPriorityComponent
 from miplearn.problems.knapsack import KnapsackInstance
 from miplearn.solvers import GurobiSolver