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)