Merge pull request #23 from EinarBaldvin/eb_dev_rrvgo

Fixed hardcoding of column names in getGoSize function.
ssayols · Mar 11, 2024 · 4f2ab8a · 4f2ab8a
2 parents b51ecb7 + eae9603
commit 4f2ab8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/rrvgo.R b/R/rrvgo.R
@@ -211,7 +211,7 @@ getGoSize <- function(terms, orgdb, keytype, children) {
   go <- AnnotationDbi::select(orgdb, keytype=if(children) "GOALL" else "GO", keys=terms, columns=keytype)
 
   # count
-  counts   <- tapply(go$ENTREZID, go$GO, function(x) length(unique(x)))
+  counts   <- tapply(go[[keytype]], go[[if(children) "GOALL" else "GO"]], function(x) length(unique(x)))
   empty    <- terms[!(terms %in% names(counts))]
   nocounts <- setNames(rep(0, length(empty)), empty)