SpanGroup(s)-related optimizations

spacy/tokens/_dict_proxies.py

@@ -42,7 +42,8 @@ def _make_span_group(self, name: str, spans: Iterable["Span"]) -> SpanGroup:
  def copy(self, doc: Optional["Doc"] = None) -> "SpanGroups":
  if doc is None:
  doc = self._ensure_doc()
- return SpanGroups(doc).from_bytes(self.to_bytes())
+ data_copy = ((k, v.copy(doc=doc)) for k, v in self.items())
+ return SpanGroups(doc, items=data_copy)
 
  def setdefault(self, key, default=None):
  if not isinstance(default, SpanGroup):

spacy/tokens/span_group.pyi

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Iterable
+from typing import Any, Dict, Iterable, Optional
 from .doc import Doc
 from .span import Span
 
@@ -24,4 +24,4 @@ class SpanGroup:
  def __getitem__(self, i: int) -> Span: ...
  def to_bytes(self) -> bytes: ...
  def from_bytes(self, bytes_data: bytes) -> SpanGroup: ...
- def copy(self) -> SpanGroup: ...
+ def copy(self, doc: Optional[Doc] = ...) -> SpanGroup: ...

spacy/tokens/span_group.pyx

@@ -241,15 +241,18 @@ cdef class SpanGroup:
  cdef void push_back(self, SpanC span) nogil:
  self.c.push_back(span)
 
- def copy(self)  -> SpanGroup:
+ def copy(self, doc: Optional["Doc"] = None) -> SpanGroup:
  """Clones the span group.
 
+ doc (Doc): New reference document to which the copy is bound.
  RETURNS (SpanGroup): A copy of the span group.
 
  DOCS: https://spacy.io/api/spangroup#copy
  """
+ if doc is None:
+ doc = self.doc
  return SpanGroup(
- self.doc,
+ doc,
  name=self.name,
  attrs=deepcopy(self.attrs),
  spans=list(self),

website/docs/api/spangroup.md

@@ -255,9 +255,10 @@ Return a copy of the span group.
 > new_group = doc.spans["errors"].copy()
 > ```
 
-| Name | Description |
-| ----------- | ----------------------------------------------- |
-| **RETURNS** | A copy of the `SpanGroup` object. ~~SpanGroup~~ |
+| Name | Description |
+| ----------- | --------------------------------------------------------------------------------------------------------------- |
+| `doc` | New reference document to which the copy is bound. Defaults to `None`. ~~Optional[Doc]~~ |
+| **RETURNS** | A copy of the `SpanGroup` object, bound to either its original reference document or the new one. ~~SpanGroup~~ |
 
 ## SpanGroup.to_bytes {#to_bytes tag="method"}