From 974cf070ce4caf1cf5135bb66a7bdc9df47f7838 Mon Sep 17 00:00:00 2001 From: Joe Rattazzi Date: Thu, 13 Feb 2025 22:34:32 -0600 Subject: [PATCH] fix merge issue --- graphrag/index/update/entities.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/graphrag/index/update/entities.py b/graphrag/index/update/entities.py index f83a6aecae..2f93769e0a 100644 --- a/graphrag/index/update/entities.py +++ b/graphrag/index/update/entities.py @@ -158,7 +158,7 @@ async def process_in_batches(df, batch_size): for start in range(0, len(df), batch_size): end = start + batch_size batch = df.iloc[start:end] - tasks = [process_row(row) for row in entities_df.itertuples(index=False, name="Entity")] + tasks = [process_row(row) for row in batch.itertuples(index=False, name="Entity")] batch_results = await asyncio.gather(*tasks) results.extend(batch_results) batch_num += 1