Branching: read priorities as float

miplearn/branching.py

@@ -42,7 +42,8 @@ class BranchPriorityComponent(Component):
                            check=True,
                            capture_output=True)
             self._merge(np.genfromtxt(priority_file.name,
-                                      delimiter=','))
+                                      delimiter=',',
+                                      dtype=np.float64))
         
     
     def fit(self, solver):

miplearn/tests/test_solver.py

@@ -56,7 +56,7 @@ def test_parallel_solve():
 def test_solver_random_branch_priority():
     instance = _get_instance()
     components = {
-        "warm-start": BranchPriorityComponent(initial_priority=np.array([1, 2, 3, 4])),
+        "warm-start": BranchPriorityComponent(initial_priority=np.array([1., 2., 3., 4.])),
     }
     solver = LearningSolver(components=components)
     solver.solve(instance)