add base angle

angle_emb/angle.py

@@ -29,6 +29,7 @@
 )
 from peft.tuners.lora import LoraLayer
 
+from .base import AngleBase
 from .utils import logger
 from .evaluation import CorrelationEvaluator
 
@@ -994,7 +995,7 @@ def __call__(self,
         return loss
 
 
-class AnglE:
+class AnglE(AngleBase):
     """
     AnglE. Everything is here👋
 

angle_emb/base.py

@@ -0,0 +1,12 @@
+from abc import ABCMeta, abstractmethod
+
+
+class AngleBase(metaclass=ABCMeta):
+
+    @abstractmethod
+    def encode(self):
+        raise NotImplementedError
+
+    @abstractmethod
+    def fit(self):
+        raise NotImplementedError

angle_emb/evaluation.py

@@ -12,6 +12,8 @@
 )
 from scipy.stats import pearsonr, spearmanr
 
+from .base import AngleBase
+
 
 class CorrelationEvaluator(object):
     def __init__(
@@ -27,7 +29,7 @@ def __init__(
         self.labels = labels
         self.batch_size = batch_size
 
-    def __call__(self, model, **kwargs) -> dict:
+    def __call__(self, model: AngleBase, **kwargs) -> dict:
         """ Evaluate the model on the given dataset.
 
         :param model: AnglE, the model to evaluate.