From 1d5599cd28f59ac418810343f269886b863cb055 Mon Sep 17 00:00:00 2001
From: Matthew Honnibal <honnibal+gh@gmail.com>
Date: Wed, 8 Nov 2017 12:18:32 +0100
Subject: [PATCH] Fix dtype

---
 spacy/_ml.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spacy/_ml.py b/spacy/_ml.py
index 171945ea2..246979a4f 100644
--- a/spacy/_ml.py
+++ b/spacy/_ml.py
@@ -94,7 +94,7 @@ def _zero_init(model):
 def _preprocess_doc(docs, drop=0.):
     keys = [doc.to_array([LOWER]) for doc in docs]
     ops = Model.ops
-    lengths = ops.asarray([arr.shape[0] for arr in keys], dtype='int32')
+    lengths = ops.asarray([arr.shape[0] for arr in keys], dtype='int64')
     keys = ops.xp.concatenate(keys)
     vals = ops.allocate(keys.shape[0]) + 1
     return (keys, vals, lengths), None