diff --git a/cohd/cohd_trapi_14.py b/cohd/cohd_trapi_14.py index 8999008..d12b0a4 100644 --- a/cohd/cohd_trapi_14.py +++ b/cohd/cohd_trapi_14.py @@ -1313,14 +1313,15 @@ def _add_kg_edge(self, node_1, node_2, cohd_result): # Add source retrieval sources = [ - { - 'resource_id': CohdTrapi._INFORES_ID, - 'resource_role': 'primary_knowledge_source', - }, { 'resource_id': 'infores:columbia-cdw-ehr-data', 'resource_role': 'supporting_data_source', }, + { + 'resource_id': CohdTrapi._INFORES_ID, + 'resource_role': 'primary_knowledge_source', + 'upstream_resource_ids': ['infores:columbia-cdw-ehr-data'] + }, ] # Add properties from COHD results to the edge attributes