From 91b111467bf9afaec93d4c3de86c8a1877587783 Mon Sep 17 00:00:00 2001
From: Paul O'Leary McCann <polm@dampfkraft.com>
Date: Mon, 17 May 2021 14:52:30 +0900
Subject: [PATCH] Minor fixes

---
 spacy/ml/models/coref_util.py | 3 ++-
 spacy/pipeline/coref.py       | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/spacy/ml/models/coref_util.py b/spacy/ml/models/coref_util.py
index 7c44692c3..6ce5127b0 100644
--- a/spacy/ml/models/coref_util.py
+++ b/spacy/ml/models/coref_util.py
@@ -1,6 +1,7 @@
 from thinc.types import Ints2d
 from spacy.tokens import Doc
-from typing import List, Tuple
+from typing import List, Tuple, Callable
+from ...util import registry
 
 # type alias to make writing this less tedious
 MentionClusters = List[List[Tuple[int, int]]]
diff --git a/spacy/pipeline/coref.py b/spacy/pipeline/coref.py
index d0fecf519..ac225d677 100644
--- a/spacy/pipeline/coref.py
+++ b/spacy/pipeline/coref.py
@@ -73,7 +73,7 @@ def make_coref(
 ) -> "CoreferenceResolver":
     """Create a CoreferenceResolver component."""
 
-    return CoreferenceResolver(nlp.vocab, model, name, span_cluster_prefix)
+    return CoreferenceResolver(nlp.vocab, model, name, span_cluster_prefix=span_cluster_prefix)
 
 
 class CoreferenceResolver(TrainablePipe):
@@ -88,7 +88,7 @@ class CoreferenceResolver(TrainablePipe):
         model: Model,
         name: str = "coref",
         *,
-        span_mentions: str,
+        span_mentions: str = "coref_mentions",
         span_cluster_prefix: str,
     ) -> None:
         """Initialize a coreference resolution component.