Protection against rare events of null dimension products

loosolab · Mar 7, 2017 · ec8a32d · ec8a32d
1 parent 6614cc7
commit ec8a32d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/R/pid.R b/R/pid.R
@@ -65,5 +65,10 @@ specific_information <- function(probabilities, source="x1", target="z") {
   term1 <- sweep(apply(probabilities, c(source, target), sum), 2, apply(probabilities, target, sum), "/")
   term2 <- log2(apply(probabilities, c(source,target), sum) / outer_prod)
 
-  colSums(apply(term1 * term2, 2, function(x) {x[is.nan(x)] <- 0; x}))
+  prod <- apply(term1 * term2, 2, function(x) {x[is.nan(x)] <- 0; x})
+  if(is.null(dim(prod))) {
+    return(prod)
+  } else {
+    return(colSums(prod))
+  }
 }