init dependency

ZiyiXia · ZiyiXia · commit 5a04983479f3 · 2024-11-07T15:25:14.000Z
diff --git a/FlagEmbedding/finetune/reranker/encoder_only/base/__init__.py b/FlagEmbedding/finetune/reranker/encoder_only/base/__init__.py
@@ -1,7 +1,9 @@
 from .modeling import CrossEncoderModel
 from .runner import EncoderOnlyRerankerRunner
+from .trainer import EncoderOnlyRerankerTrainer
 
 __all__ = [
     "CrossEncoderModel",
-    "EncoderOnlyRerankerRunner"
+    "EncoderOnlyRerankerRunner",
+    "EncoderOnlyRerankerTrainer"
 ]
