Merge branch 'partitioner' of https://github.com/open-sciencelab/GraphGen into partitioner

Author: ChenZiHong-Gavin

graphgen/models/generator/atomic_generator.py

@@ -1,6 +1,6 @@
 from typing import Any
 
-from baselines.EntiGraph.tasks.baseline_task import compute_content_hash
+from graphgen.utils import compute_content_hash
 from graphgen.bases import BaseGenerator
 from graphgen.templates import ATOMIC_GENERATION_PROMPT
 from graphgen.utils import detect_main_language, logger
@@ -36,7 +36,7 @@ def parse_response(self, response: str) -> dict:
             answer = response.split("答案：")[1].strip()
         else:
             logger.warning("Failed to parse response: %s", response)
-            return None, None
+            return {}
         question = question.strip('"')
         answer = answer.strip('"')
         logger.info("Question: %s", question)

tests/integration_tests/models/partitioner/test_dfs_partitioner.py

@@ -48,7 +48,7 @@ async def test_small_graph():
     0 - 1 - 2
     |   |   |
     3 - 4 - 5
-    6 nodes & 7 edged，max_units=4 => 3 communities
+    6 nodes & 7 edges，max_units=4 => 3 communities
     """
     nodes = [(str(i), {"desc": f"node{i}"}) for i in range(6)]
     edges = [