fix: fix lint problem

Author: ChenZiHong-Gavin

graphgen/models/generator/cot_generator.py

@@ -3,7 +3,7 @@
 
 from graphgen.bases import BaseGenerator
 from graphgen.templates import COT_GENERATION_PROMPT
-from graphgen.utils import compute_content_hash, detect_main_language, logger
+from graphgen.utils import detect_main_language, logger
 
 
 class CoTGenerator(BaseGenerator):

graphgen/models/generator/fill_in_blank_generator.py

@@ -3,7 +3,7 @@
 
 from graphgen.bases import BaseGenerator
 from graphgen.templates import FILL_IN_BLANK_GENERATION_PROMPT
-from graphgen.utils import compute_content_hash, detect_main_language, logger
+from graphgen.utils import detect_main_language, logger
 
 
 class FillInBlankGenerator(BaseGenerator):

graphgen/models/generator/multi_answer_generator.py

@@ -3,7 +3,7 @@
 
 from graphgen.bases import BaseGenerator
 from graphgen.templates import MAQ_GENERATION_PROMPT
-from graphgen.utils import compute_content_hash, detect_main_language, logger
+from graphgen.utils import detect_main_language, logger
 
 
 class MultiAnswerGenerator(BaseGenerator):

graphgen/models/generator/multi_choice_generator.py

@@ -3,7 +3,7 @@
 
 from graphgen.bases import BaseGenerator
 from graphgen.templates import MCQ_GENERATION_PROMPT
-from graphgen.utils import compute_content_hash, detect_main_language, logger
+from graphgen.utils import detect_main_language, logger
 
 
 class MultiChoiceGenerator(BaseGenerator):

graphgen/models/generator/multi_hop_generator.py

@@ -3,7 +3,7 @@
 
 from graphgen.bases import BaseGenerator
 from graphgen.templates import MULTI_HOP_GENERATION_PROMPT
-from graphgen.utils import compute_content_hash, detect_main_language, logger
+from graphgen.utils import detect_main_language, logger
 
 
 class MultiHopGenerator(BaseGenerator):

graphgen/models/generator/vqa_generator.py

@@ -3,7 +3,7 @@
 
 from graphgen.bases import BaseGenerator
 from graphgen.templates import VQA_GENERATION_PROMPT
-from graphgen.utils import compute_content_hash, detect_main_language, logger
+from graphgen.utils import detect_main_language, logger
 
 
 class VQAGenerator(BaseGenerator):
@@ -83,21 +83,21 @@ async def generate(
 
     @staticmethod
     def format_generation_results(
-        results: list[dict], output_data_format: str
+        result: list[dict], output_data_format: str
     ) -> list[dict[str, Any]]:
         if output_data_format == "Alpaca":
-            results = [
+            result = [
                 {
                     "instruction": v["question"],
                     "input": "",
                     "output": v["answer"],
                     "image": v.get("img_path", ""),
                 }
-                for item in results
+                for item in result
                 for k, v in item.items()
             ]
         elif output_data_format == "Sharegpt":
-            results = [
+            result = [
                 {
                     "conversations": [
                         {
@@ -109,11 +109,11 @@ def format_generation_results(
                         {"from": "gpt", "value": [{"text": v["answer"]}]},
                     ]
                 }
-                for item in results
+                for item in result
                 for k, v in item.items()
             ]
         elif output_data_format == "ChatML":
-            results = [
+            result = [
                 {
                     "messages": [
                         {
@@ -128,9 +128,9 @@ def format_generation_results(
                         },
                     ]
                 }
-                for item in results
+                for item in result
                 for k, v in item.items()
             ]
         else:
             raise ValueError(f"Unknown output data format: {output_data_format}")
-        return results
+        return result