#321 actually this one for the predicate type issue

RTXteam · Aug 3, 2023 · 2cad675 · 2cad675
1 parent 5f056f4
commit 2cad675
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/report_stats_on_kg_jsonl.py b/report_stats_on_kg_jsonl.py
@@ -52,7 +52,7 @@ def get_edge_stats(edges_file_name: list):
     subject_key = 'subject'
     object_key = 'object'
     predicate_curie_key = 'source_predicate' if not args.use_simplified_predicates else 'predicate'
-    label_key = 'relation_label' if not args.use_simplified_predicates else 'predicate'
+    label_key = 'relation_label' if not args.use_simplified_predicates else 'predicate_label'
 
     edge_count = 0
     edge_sources = dict()