Force TFP modules to load, fix model init

liesel-devs · Jun 26, 2024 · badb029 · badb029
1 parent 724d32e
commit badb029
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/R/utils.R b/R/utils.R
@@ -52,6 +52,9 @@ plot.liesel.model.model.Model <- function(x, nodes = FALSE, ...) {
 #' @importFrom reticulate py_has_attr
 
 get_distribution <- function(x) {
+  .lsld$GaussianCopula  # fix #4
+  .tfd$Normal  # fix #4
+
   if (is.character(x)) {
     if (py_has_attr(.lsld, x)) .lsld[[x]] else .tfd[[x]]
   } else {
@@ -63,6 +66,9 @@ get_distribution <- function(x) {
 #' @importFrom reticulate py_has_attr
 
 get_bijector <- function(x) {
+  .lslb$AlgebraicSigmoid  # fix #4
+  .tfb$Identity  # fix #4
+
   if (is.character(x)) {
     if (py_has_attr(.lslb, x)) .lslb[[x]] else .tfb[[x]]
   } else {