* Allow slice indexing in Doc.__getitem__, returning a Span object

2015-07-09 15:15:32 +02:00 · 2015-07-09 15:15:32 +02:00 · c0255ed7d8
parent 7d2964f673
commit c0255ed7d8
1 changed files with 6 additions and 0 deletions
--- a/spacy/tokens.pyx
+++ b/spacy/tokens.pyx
@ -115,6 +115,12 @@ cdef class Doc:
        Returns:
            token (Token):
        """
+        if isinstance(i, slice):
+            if i.step is not None:
+                raise ValueError("Stepped slices not supported in Span objects."
+                                 "Try: list(doc)[start:stop:step] instead.")
+            return Span(self, i.start, i.stop, label=0)
+
        if i < 0:
            i = self.length + i
        bounds_check(i, self.length, PADDING)