diff --git a/spacy/tokens/doc.pyx b/spacy/tokens/doc.pyx index 66936c4a5..ce2a82cd0 100644 --- a/spacy/tokens/doc.pyx +++ b/spacy/tokens/doc.pyx @@ -499,10 +499,10 @@ cdef class Doc: """ cdef int i, j cdef attr_id_t feature - cdef np.ndarray[attr_t, ndim=1] attr_ids, output_1D + cdef np.ndarray[attr_t, ndim=1] attr_ids cdef np.ndarray[attr_t, ndim=2] output # Handle scalar/list inputs of strings/ints for py_attr_ids - if not hasattr(py_attr_ids, '__iter__'): + if not hasattr(py_attr_ids, '__iter__'): py_attr_ids = [py_attr_ids] # Allow strings, e.g. 'lemma' or 'LEMMA'