mirror of
https://github.com/microsoft/graphrag.git
synced 2026-01-14 00:57:23 +08:00
Add empty objects for NaN raw_data
This commit is contained in:
parent
7ce10306ee
commit
e170124d39
@ -67,4 +67,7 @@ def create_final_documents(
|
||||
rejoined["id"] = rejoined["id"].astype(str)
|
||||
rejoined["human_readable_id"] = rejoined.index
|
||||
|
||||
if "raw_data" not in rejoined.columns:
|
||||
rejoined["raw_data"] = pd.Series(dtype="object")
|
||||
|
||||
return rejoined.loc[:, DOCUMENTS_FINAL_COLUMNS]
|
||||
|
||||
Loading…
Reference in New Issue
Block a user