Fix function for one-sample cases.

charles-plessy · Jun 28, 2023 · a019141 · a019141
1 parent 8b70db0
commit a019141
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/R/ImportMethods.R b/R/ImportMethods.R
@@ -796,10 +796,14 @@ setGeneric("importPublicData",
                      df$strand,
                      seqinfo = seqinfo(getRefGenome(genome.name)),
                      bsgenomeName = genome.name)
-  ctssDF <- lapply(df[ , sample], Rle) |> DataFrame()
+  ctssDF <- lapply(df[ , sample, drop = FALSE], Rle) |> DataFrame()
   colnames(ctssDF) <- make.names(colnames(ctssDF))
   ctssSE <- SummarizedExperiment(c(counts = ctssDF), ctssRanges)
-  ctssSE <- ctssSE[rowSums(df[ , sample]) > 0,]
+  if (ncol(ctssDF) == 1) {
+    ctssSE <- ctssSE[        df[ , sample]  > 0,]
+  } else {
+    ctssSE <- ctssSE[rowSums(df[ , sample]) > 0,]
+  }
   sort(ctssSE)
 }