From 5d6995d3178f9843967a1f11bb11bb08ab768097 Mon Sep 17 00:00:00 2001 From: Cosimo Lupo Date: Thu, 19 Jan 2017 15:39:48 +0000 Subject: [PATCH] [Snippets/interpolatable] scipy returns numpy arrays; convert them back to list --- Snippets/interpolatable.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/Snippets/interpolatable.py b/Snippets/interpolatable.py index 81bafb1c6..4b78180f8 100755 --- a/Snippets/interpolatable.py +++ b/Snippets/interpolatable.py @@ -80,9 +80,8 @@ def min_cost_perfect_bipartite_matching(G): try: from scipy.optimize import linear_sum_assignment rows, cols = linear_sum_assignment(G) - # This branch untested - assert rows == list(range(n)) - return cols, _matching_cost(G, cols) + assert (rows == list(range(n))).all() + return list(cols), _matching_cost(G, cols) except ImportError: pass