From ce5d133989c507cb7645ff3789645e0564a58c7b Mon Sep 17 00:00:00 2001 From: Martin Mauch Date: Sun, 24 Nov 2019 22:16:58 +0100 Subject: [PATCH] Always return ints in path --- dtw/dtw.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dtw/dtw.py b/dtw/dtw.py index 48d6dfc..1e98f91 100644 --- a/dtw/dtw.py +++ b/dtw/dtw.py @@ -47,9 +47,9 @@ def dtw(x, y, dist, warp=1, w=inf, s=1.0): min_list += [D0[i_k, j] * s, D0[i, j_k] * s] D1[i, j] += min(min_list) if len(x) == 1: - path = zeros(len(y)), range(len(y)) + path = zeros(len(y),dtype=int), range(len(y)) elif len(y) == 1: - path = range(len(x)), zeros(len(x)) + path = range(len(x)), zeros(len(x),dtype=int) else: path = _traceback(D0) return D1[-1, -1], C, D1, path