Amend 8ae8b443f: Handle comparison with None tokens.

2017-01-11 13:03:32 +01:00 · 2017-01-11 13:03:32 +01:00 · 12cd27b821
parent 44e2b0100d
commit 12cd27b821
1 changed files with 2 additions and 2 deletions
--- a/spacy/tokens/token.pyx
+++ b/spacy/tokens/token.pyx
@ -60,10 +60,10 @@ cdef class Token:
    def __repr__(self):
        return self.__str__()

-    def __richcmp__(self, other, int op):
+    def __richcmp__(self, Token other, int op):
        # http://cython.readthedocs.io/en/latest/src/userguide/special_methods.html
        my = self.idx
-        their = other.idx
+        their = other.idx if other is not None else None
        if op == 0:
            return my < their
        elif op == 2: