Fix optional covariates check in incremental indexing (#1374)

AlonsoGuevara · web-flow · commit 1661672569a2 · 2024-11-06T17:22:11.000-06:00
* Fix optional covariates check in incremental indexing

* Oopsie fix
diff --git a/.semversioner/next-release/patch-20241106225803494336.json b/.semversioner/next-release/patch-20241106225803494336.json
@@ -0,0 +1,4 @@
+{
+  "type": "patch",
+  "description": "Fix optional covariates update in incremental indexing"
+}
diff --git a/graphrag/index/update/incremental_index.py b/graphrag/index/update/incremental_index.py
@@ -120,8 +120,12 @@ async def update_dataframe_outputs(
     )
 
     # Merge final covariates
-    progress_reporter.info("Updating Final Covariates")
-    await _update_covariates(dataframe_dict, storage, update_storage)
+    if (
+        await storage.has("create_final_covariates.parquet")
+        and "create_final_covariates" in dataframe_dict
+    ):
+        progress_reporter.info("Updating Final Covariates")
+        await _update_covariates(dataframe_dict, storage, update_storage)
 
     # Merge final nodes and update community ids
     progress_reporter.info("Updating Final Nodes")

-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +{
 +  "type": "patch",
 +  "description": "Fix optional covariates update in incremental indexing"
 +}