Added custom parameters(<dist_wrap>) method to drop fn env

mitchelloharawild · Nov 5, 2021 · eb5a6d3 · eb5a6d3
1 parent 01d96c8
commit eb5a6d3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -325,6 +325,7 @@ S3method(mean,distribution)
 S3method(median,dist_default)
 S3method(median,distribution)
 S3method(parameters,dist_default)
+S3method(parameters,dist_wrap)
 S3method(parameters,distribution)
 S3method(print,dist_bernoulli)
 S3method(print,dist_beta)

diff --git a/R/dist_wrap.R b/R/dist_wrap.R
@@ -112,3 +112,9 @@ generate.dist_wrap <- function(x, times, ...){
 
   do.call(fn, c(list(times), par))
 }
+
+#' @export
+parameters.dist_wrap <- function(x, ...) {
+  # All parameters except distribution environment
+  x[-2]
+}