Update cohort-diagnostics-main.R

adding check for column names used in filter and mutate to cohort-diagnostics
OHDSI · Oct 2, 2024 · b5cb001 · b5cb001
1 parent 942c6ea
commit b5cb001
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/R/cohort-diagnostics-main.R b/R/cohort-diagnostics-main.R
@@ -621,6 +621,9 @@ cohortDiagnosticsServer <- function(id,
       if (!hasData(targetCohortId())) {
         return(NULL)
       }
+      if (sum(c('cohortId','conceptSetName') %in% colnames(dataSource$conceptSets)) !=2) {
+        return(NULL)
+      }
       dataSource$conceptSets %>%
         dplyr::filter(.data$cohortId == targetCohortId()) %>%
         dplyr::mutate(name = .data$conceptSetName) %>%