solve circular import

SeanLee97 · Jul 26, 2024 · ddd97b6 · ddd97b6
1 parent c51e05d
commit ddd97b6
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/angle_emb/evaluation.py b/angle_emb/evaluation.py
@@ -12,8 +12,6 @@
 )
 from scipy.stats import pearsonr, spearmanr
 
-from .angle import AnglE
-
 
 class CorrelationEvaluator(object):
     def __init__(
@@ -29,7 +27,7 @@ def __init__(
         self.labels = labels
         self.batch_size = batch_size
 
-    def __call__(self, model: AnglE, **kwargs) -> dict:
+    def __call__(self, model, **kwargs) -> dict:
         """ Evaluate the model on the given dataset.
 
         :param model: AnglE, the model to evaluate.