Attempting to fix Windows CI

rth · rth · commit 121a95de24a3 · 2017-11-11T23:40:52.000+01:00
diff --git a/sklearn/feature_extraction/_hashing.pyx b/sklearn/feature_extraction/_hashing.pyx
@@ -1,6 +1,8 @@
 # Author: Lars Buitinck
 # License: BSD 3 clause
 
+import os
+import sys
 import array
 from cpython cimport array
 cimport cython
@@ -34,12 +36,19 @@ def transform(raw_X, Py_ssize_t n_features, dtype, bint alternate_sign=1):
     cdef array.array indices
     cdef array.array indptr
     indices = array.array("i")
-    indptr = array.array("l", [0])
+    if sys.version_info >= (3, 3):
+        indptr = array.array("q", [0])
+    elif os.name == 'nt':
+        # Win PY2.7
+        indptr = array.array("i", [0])
+    else:
+        # Linux PY2.7
+        indptr = array.array("l", [0])
 
     # Since Python array does not understand Numpy dtypes, we grow the indices
     # and values arrays ourselves. Use a Py_ssize_t capacity for safety.
     cdef Py_ssize_t capacity = 8192     # arbitrary
-    cdef np.intp_t size = 0
+    cdef np.int64_t size = 0
     cdef np.ndarray values = np.empty(capacity, dtype=dtype)
 
     for x in raw_X:
