Merge pull request #181 from WengLab-InformaticsResearch/upstream_res…

…ource_ids Add upstream_resource_ids on COHD sources
WengLab-InformaticsResearch · Aug 11, 2023 · 5b8bf13 · 5b8bf13
2 parents 9d5769c + 81f79fc
commit 5b8bf13
Showing 1 changed file with 5 additions and 4 deletions.
diff --git 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