From 65e84b56c043032a45f7cf2cbf4188b75c11d72a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Laura=20Kl=C3=BCnder?= Date: Tue, 26 Dec 2017 15:53:08 +0100 Subject: [PATCH] don't crash with index stuff while locating --- src/c3nav/routing/locator.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/c3nav/routing/locator.py b/src/c3nav/routing/locator.py index 87381346..44e472c2 100644 --- a/src/c3nav/routing/locator.py +++ b/src/c3nav/routing/locator.py @@ -150,8 +150,9 @@ class LocatorSpace: scores = np.sum((self.levels[np.array(acceptable_points, dtype=np.uint32).reshape((-1, 1)), stations]-values)**2, axis=1) / len(stations) - best_point = acceptable_points[np.argmin(scores).ravel()[0]] - return self.points[best_point], scores[best_point] + best_acceptable_point = np.argmin(scores).ravel()[0] + best_point = acceptable_points[best_acceptable_point] + return self.points[best_point], scores[best_acceptable_point] class LocatorPoint(namedtuple('LocatorPoint', ('x', 'y', 'values'))):