Adds support for CHATTR_MODEL

mlverse · Apr 10, 2024 · 86c1681 · 86c1681
1 parent 95b0404
commit 86c1681
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/R/chattr-defaults.R b/R/chattr-defaults.R
@@ -75,20 +75,23 @@ chattr_defaults <- function(type = "default",
   if (force) {
     ch_env$defaults <- NULL
   }
-
+  env_model <- NULL
   if (is.null(chattr_defaults_get(type))) {
-
     # Overrides environment variable if YAML file is present
     if (file_exists(yaml_file)) {
       check_files <- yaml_file
     } else {
       check_files <- ch_package_file(Sys.getenv("CHATTR_USE", unset = NA))
     }
-
+    env_model <- Sys.getenv("CHATTR_MODEL", unset = NA)
+    if(is.na(env_model)) {
+      env_model <- NULL
+    }
     for (j in seq_along(check_files)) {
       td_defaults <- read_yaml(file = check_files[j])
       loaded_default <- chattr_defaults_get(type = "default")
       td_defaults$default <- loaded_default %||% td_defaults$default
+      td_defaults$default[["model"]] <- env_model %||% td_defaults$default[["model"]]
       check_defaults <- c("default", type)
       for (i in seq_along(check_defaults)) {
         td <- td_defaults[[check_defaults[i]]]

diff --git a/R/chattr-use.R b/R/chattr-use.R
@@ -155,7 +155,9 @@ use_switch <- function(..., .file) {
 }
 
 ch_package_file <- function(x) {
-  if(is.na(x)) return(NULL)
+  if (is.na(x)) {
+    return(NULL)
+  }
   env_folder <- ifelse(x == "test", "apptest", "configs")
   package_file(env_folder, path_ext_set(x, "yml"))
 }