From e8df283b247298216dc32e90728001dd6b8e2915 Mon Sep 17 00:00:00 2001 From: Kapil Sachdeva Date: Wed, 24 Jul 2024 14:26:37 -0500 Subject: [PATCH] fix - in run_local_search(), avoid re-reading create_final_nodes.parquet file (#592) --- .semversioner/next-release/patch-20240724150548966311.json | 4 ++++ graphrag/query/cli.py | 1 - 2 files changed, 4 insertions(+), 1 deletion(-) create mode 100644 .semversioner/next-release/patch-20240724150548966311.json diff --git a/.semversioner/next-release/patch-20240724150548966311.json b/.semversioner/next-release/patch-20240724150548966311.json new file mode 100644 index 00000000..36487847 --- /dev/null +++ b/.semversioner/next-release/patch-20240724150548966311.json @@ -0,0 +1,4 @@ +{ + "type": "patch", + "description": "remove duplicate file read" +} diff --git a/graphrag/query/cli.py b/graphrag/query/cli.py index 306dad60..10ad95d5 100644 --- a/graphrag/query/cli.py +++ b/graphrag/query/cli.py @@ -110,7 +110,6 @@ def run_local_search( final_relationships = pd.read_parquet( data_path / "create_final_relationships.parquet" ) - final_nodes = pd.read_parquet(data_path / "create_final_nodes.parquet") final_entities = pd.read_parquet(data_path / "create_final_entities.parquet") final_covariates_path = data_path / "create_final_covariates.parquet" final_covariates = (