Skip to content

Commit b2db994

Browse files
Merge branch 'feature/vqa-pipeline' of https://github.com/open-sciencelab/GraphGen into feature/vqa-pipeline
2 parents ef2e109 + d5bbdcb commit b2db994

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

graphgen/operators/partition/partition_kg.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ async def partition_kg(
6060
nodes, edges = batch
6161
for node_id, node_data in nodes:
6262
entity_type = node_data.get("entity_type")
63-
if "image" in entity_type.lower():
63+
if entity_type and "image" in entity_type.lower():
6464
node_id = node_id.strip('"').lower()
6565
image_data = await chunk_storage.get_by_id(node_id)
6666
if image_data:

0 commit comments

Comments
 (0)