Merge pull request #21 from vene/py3

FIX py3 compat in transmatrix
larsmans · Feb 16, 2016 · 32d4bfa · 32d4bfa
2 parents b85c56e + 1493d32
commit 32d4bfa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/seqlearn/_utils/transmatrix.py b/seqlearn/_utils/transmatrix.py
@@ -3,6 +3,7 @@
 
 import numpy as np
 from scipy.sparse import csr_matrix
+from sklearn.externals import six
 
 
 def make_trans_matrix(y, n_classes, dtype=np.float64):
@@ -14,7 +15,7 @@ def make_trans_matrix(y, n_classes, dtype=np.float64):
     """
     indices = np.empty(len(y), dtype=np.int32)
 
-    for i in xrange(len(y) - 1):
+    for i in six.moves.xrange(len(y) - 1):
         indices[i] = y[i] * i + y[i + 1]
 
     indptr = np.arange(len(y) + 1)