NiFi: cohort script update.

CogStack · Jul 8, 2024 · e087951 · e087951
1 parent 5460622
commit e087951
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nifi/user-scripts/cogstack_cohort_generate_data.py b/nifi/user-scripts/cogstack_cohort_generate_data.py
@@ -173,7 +173,7 @@ def _process_annotation_records(annotation_records: list):
                 annotation_entity = annotation_record["_source"]
             docid = annotation_entity[ANNOTATION_DOCUMENT_ID_FIELD_NAME]
 
-            if str(docid) in unique_doc_ids.keys():
+            if str(docid) in unique_doc_ids:
                 patient_id = global_doc2ptt[str(docid)]
                 cui = annotation_entity["nlp.cui"]