From b1118cee584d8a1a4eb40dfd6d9660388807cf12 Mon Sep 17 00:00:00 2001
From: Paul O'Leary McCann <polm@dampfkraft.com>
Date: Tue, 24 May 2022 15:59:08 +0900
Subject: [PATCH] Move epsilon

---
 spacy/ml/models/coref.py      | 1 +
 spacy/ml/models/coref_util.py | 2 --
 2 files changed, 1 insertion(+), 2 deletions(-)

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