diff --git a/spacy/_ml.py b/spacy/_ml.py index 5d388913b..3db28ced5 100644 --- a/spacy/_ml.py +++ b/spacy/_ml.py @@ -360,7 +360,7 @@ def Tok2Vec(width, embed_size, **kwargs): "config": { "vectors_name": pretrained_vectors, "width": width, - "column": cols.index(ID), + "column": cols.index("ID") }, } if cnn_maxout_pieces >= 2: