diff --git a/spacy/ml/models/coref.py b/spacy/ml/models/coref.py index 0667053c6..b4d8030e8 100644 --- a/spacy/ml/models/coref.py +++ b/spacy/ml/models/coref.py @@ -185,6 +185,7 @@ class CorefScorer(torch.nn.Module): return coref_scores, top_indices +EPSILON = 1e-7 # Note this function is kept here to keep a torch dep out of coref_util. def add_dummy(tensor: torch.Tensor, eps: bool = False): """Prepends zeros (or a very small value if eps is True) diff --git a/spacy/ml/models/coref_util.py b/spacy/ml/models/coref_util.py index 05f83189a..86dd0df4b 100644 --- a/spacy/ml/models/coref_util.py +++ b/spacy/ml/models/coref_util.py @@ -8,8 +8,6 @@ MentionClusters = List[List[Tuple[int, int]]] DEFAULT_CLUSTER_PREFIX = "coref_clusters" -EPSILON = 1e-7 - class GraphNode: def __init__(self, node_id: int): self.id = node_id