From 354d0ac541ed9a7f686f0831aede7e152d75a44b Mon Sep 17 00:00:00 2001 From: Alinson S Xavier Date: Thu, 30 Jan 2020 13:46:45 -0600 Subject: [PATCH] Branching: read priorities as float --- miplearn/branching.py | 3 ++- miplearn/tests/test_solver.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/miplearn/branching.py b/miplearn/branching.py index 40c2561..0654953 100644 --- a/miplearn/branching.py +++ b/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): diff --git a/miplearn/tests/test_solver.py b/miplearn/tests/test_solver.py index 2e97fab..fc7c532 100644 --- a/miplearn/tests/test_solver.py +++ b/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)