Make a dtype more specific, to fix a windows build

This commit is contained in:
Matthew Honnibal 2017-11-08 11:24:35 +01:00
parent 97ed21492a
commit a8b592783b

View File

@ -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])
lengths = ops.asarray([arr.shape[0] for arr in keys], dtype='int32')
keys = ops.xp.concatenate(keys)
vals = ops.allocate(keys.shape[0]) + 1
return (keys, vals, lengths), None