fix: fix cyclic import

Author: ChenZiHong-Gavin

graphgen/operators/split_graph.py

@@ -224,7 +224,7 @@ def _sort_edges(edges: list, edge_sampling: str) -> list:
         raise ValueError(f"Invalid edge sampling: {edge_sampling}")
     return edges
 
-async def get_batches_with_strategy( # pylint: disable=too-many-arguments
+async def get_batches_with_strategy( # pylint: disable=too-many-branches
     nodes: list,
     edges: list,
     graph_storage: NetworkXStorage,

models/evaluate/uni_evaluator.py

@@ -7,7 +7,7 @@
 import torch.multiprocessing as mp
 
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-from models import TextPair
+from models.text.text_pair import TextPair
 
 
 def _add_questions(dimension: str, question: str, answer: str):

models/llm/openai_model.py

@@ -10,7 +10,7 @@
     retry_if_exception_type,
 )
 
-from models import TopkTokenModel, Token
+from models.llm.topk_token_model import TopkTokenModel, Token
 
 
 def get_top_response_tokens(response: openai.ChatCompletion) -> List[Token]: