From 150410674607a4e055b36e7eb1e857e4fa1f1052 Mon Sep 17 00:00:00 2001
From: Joshua Lambert <joshua.lambert@lshtm.ac.uk>
Date: Mon, 13 Nov 2023 16:15:18 +0000
Subject: [PATCH] update cdf.epidist to handle log prob

---
 R/epidist.R                           | 2 ++
 man/epidist_distribution_functions.Rd | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/R/epidist.R b/R/epidist.R
index 6f86e6038..0c46e9fcb 100644
--- a/R/epidist.R
+++ b/R/epidist.R
@@ -613,6 +613,7 @@ is_epidist <- function(x) {
 #' @param p The probabilities to evaluate at.
 #' @param times The number of random samples.
 #' @inheritParams print.epidist
+#' @inheritParams distributional::cdf
 #'
 #' @return If an `<epidist>` object is given a numeric vector is returned, if an
 #' `<vb_epidist>` object is given a list of two elements each with a numeric
@@ -693,6 +694,7 @@ cdf.epidist <- function(x, q, ..., log = FALSE) {
   unlist <- length(x$prob_dist) == 1
   if (inherits(x$prob_dist, "distcrete")) {
     out <- x$prob_dist$p(q)
+    if (log) out <- log(out)
   } else {
     out <- distributional::cdf(x$prob_dist, q = q, ..., log = log)
   }
diff --git a/man/epidist_distribution_functions.Rd b/man/epidist_distribution_functions.Rd
index 4dfbdd901..56ce7b789 100644
--- a/man/epidist_distribution_functions.Rd
+++ b/man/epidist_distribution_functions.Rd
@@ -38,6 +38,8 @@
 
 \item{q}{The quantiles to evaluate at.}
 
+\item{log}{If \code{TRUE}, probabilities will be given as log probabilities.}
+
 \item{p}{The probabilities to evaluate at.}
 
 \item{times}{The number of random samples.}