fix bugs

codecritic/cli/run_sft_test.py

@@ -8,9 +8,9 @@ from transformers import AutoTokenizer
 
 from codecritic.data.code import extract_code, code_template
 from codecritic.data.cov import COV_PROMPT
-from codecritic.utils.vllm import vllm_chatcomplete, vllm_score
-from codecritic.utils.json import load_jsonl, save_jsonl 
 from codecritic.data.verify import get_score_token_id
+from codecritic.utils.vllm import vllm_chatcomplete, vllm_score
+from codecritic.utils.json import load_jsonl, save_jsonl
 from codecritic.utils.metric import group_results, score_pass_at_k
 
 

codecritic/data/cov.py

@@ -3,9 +3,7 @@
 from codecritic.data.code import extract_code, code_template
 
 from codecritic.data.utils import SPLITTER, mk_message, mk_messages
-from codecritic.data.verify import (
-    mk_critic_verify,
-)
+from codecritic.data.verify import mk_critic_verify
 
 COV_PROMPT = "Please verify your code step by step using Markdown code blocks. After each step, explain whether it's correct or not, and if not, explain the issue."
 

codecritic/sampling/evaluate_code.py

@@ -9,7 +9,7 @@ from tqdm.contrib.concurrent import process_map
 
 from codecritic.sampling.apps_test import run_test
 from codecritic.utils.json import save_jsonl
-from codecritic.utils.data import extract_code
+from codecritic.data.code import extract_code
 
 TIMEOUT = 10