Handling edge case when exporting preference distribution and the tar…

…get or comparator only has 1 subject.
OHDSI · Nov 7, 2023 · c93d0fb · c93d0fb
1 parent 73e299e
commit c93d0fb
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/NEWS.md b/NEWS.md
@@ -24,7 +24,9 @@ Changes:
 
 Bugfixes:
 
-3. Fixing `runCmAnalyses()` when using `refitPsForEveryOutcome = TRUE`.
+1. Fixing `runCmAnalyses()` when using `refitPsForEveryOutcome = TRUE`.
+
+2. Handling edge case when exporting preference distribution and the target or comparator only has 1 subject.
 
 
 CohortMethod 5.1.0

diff --git a/R/Export.R b/R/Export.R
@@ -811,7 +811,10 @@ exportPreferenceScoreDistribution <- function(outputFolder,
   # rows <- split(reference, reference$sharedPsFile)[[2]]
   preparePlot <- function(rows) {
     ps <- readRDS(file.path(outputFolder, rows$sharedPsFile[1]))
-    if (nrow(ps) > 0 && min(ps$propensityScore) < max(ps$propensityScore)) {
+    if (nrow(ps) > 0 &&
+        min(ps$propensityScore) < max(ps$propensityScore) &&
+        sum(ps$treatment == 1) > 1 &&
+        sum(ps$treatment == 0) > 1 ) {
       ps <- computePreferenceScore(ps)
       d1 <- density(ps$preferenceScore[ps$treatment == 1], from = 0, to = 1, n = 100)
       d0 <- density(ps$preferenceScore[ps$treatment == 0], from = 0, to = 1, n = 100)