From a8f1466d58f80f6de0a57073a21cc21f23971449 Mon Sep 17 00:00:00 2001 From: GhostC <1276537536@qq.com> Date: Thu, 24 Jul 2025 21:34:41 +0800 Subject: [PATCH] fix: fix some obvious faults --- retriever/subgraph.py | 2 +- rewriter/inference_rewriter.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/retriever/subgraph.py b/retriever/subgraph.py index 17cbd0f..7202ad9 100644 --- a/retriever/subgraph.py +++ b/retriever/subgraph.py @@ -184,4 +184,4 @@ def get_graph_file_name(item): # save with open("subgraph_nodes.json", 'w', encoding='utf-8') as file: - json.dump(anchor_node_dict, file) \ No newline at end of file + json.dump(subgraph_id_dict, file) \ No newline at end of file diff --git a/rewriter/inference_rewriter.py b/rewriter/inference_rewriter.py index 124d2aa..e939b3b 100644 --- a/rewriter/inference_rewriter.py +++ b/rewriter/inference_rewriter.py @@ -98,8 +98,8 @@ def inference_LLM_patch(prompt_path): torch.cuda.empty_cache() #### save output #### - test_basic_info["rewriter_inferer"] = response_list[:instance_num] - test_basic_info["rewriter_extractor"] = response_list[instance_num:] + test_basic_info["rewriter_extractor"] = response_list[:instance_num] + test_basic_info["rewriter_inferer"] = response_list[instance_num:] test_basic_info.to_json("test_rewriter_output.json", index=False)