merge

Author: CHERRY-ui8

graphgen/operators/__init__.py

@@ -1,5 +1,6 @@
 from .build_kg import BuildKGService
 from .chunk import ChunkService
+from .evaluate import EvaluateService
 from .extract import ExtractService
 from .generate import GenerateService
 from .judge import JudgeService

graphgen/operators/evaluate/__init__.py

@@ -1 +1,3 @@
 from .evaluate_service import EvaluateService
+
+__all__ = ["EvaluateService"]

graphgen/operators/evaluate/evaluate_service.py

@@ -48,10 +48,6 @@ def _init_evaluators(self):
             else:
                 raise ValueError(f"Unknown metric: {metric}")
 
-    def process(self, batch: pd.DataFrame) -> pd.DataFrame:
-        items = batch.to_dict(orient="records")
-        return pd.DataFrame(self.evaluate(items))
-
     async def _process_single(self, item: dict[str, Any]) -> dict[str, Any]:
         try:
             qa_pair = QAPair(