GERO-318 - bugfix - use get for cases with no source given.

bcgsc · Apr 12, 2023 · 40fd455 · 40fd455
1 parent b8c0255
commit 40fd455
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ipr/ipr.py b/ipr/ipr.py
@@ -72,7 +72,7 @@ def get_evidencelevel_mapping(graphkb_conn: GraphKBConnection) -> Dict[str, str]
 
     # Filter IPR EvidenceLevel and map each outgoing CrossReferenceOf to displayName
     ipr_source_rid = graphkb_conn.get_source("ipr")["@rid"]
-    ipr_evidence_levels = filter(lambda d: d["source"] == ipr_source_rid, evidence_levels)
+    ipr_evidence_levels = filter(lambda d: d.get("source") == ipr_source_rid, evidence_levels)
     cross_references_mapping: Dict[str, str] = dict()
     ipr_rids_to_displayname = dict()
     for level in ipr_evidence_levels: