Rprofile unneeded, style

cmu-delphi · Jan 18, 2024 · 52f6dd1 · 52f6dd1
1 parent f96e4bf
commit 52f6dd1
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 15 deletions.
diff --git a/.Rprofile b/.Rprofile
diff --git a/R/make_quantile_reg.R b/R/make_quantile_reg.R
@@ -49,8 +49,8 @@ quantile_reg <- function(mode = "regression", engine = "rq", quantile_levels = 0
 }
 
 make_quantile_reg <- function() {
-  current_env <- get_model_env()
-  if (!any(current_env$models == "quantile_reg")) {
+  model_env <- get_model_env()
+  if (!("quantile_reg" %in% model_env$models)) {
     parsnip::set_new_model("quantile_reg")
   }
   parsnip::set_model_mode("quantile_reg", "regression")

diff --git a/R/make_smooth_quantile_reg.R b/R/make_smooth_quantile_reg.R
@@ -111,8 +111,8 @@ smooth_quantile_reg <- function(
 
 
 make_smooth_quantile_reg <- function() {
-  current_env <- get_model_env()
-  if (!any(current_env$models == "smooth_quantile_reg")) {
+  model_env <- get_model_env()
+  if (!("smooth_quantile_reg" %in% model_env$models)) {
     parsnip::set_new_model("smooth_quantile_reg")
   }
   parsnip::set_model_mode("smooth_quantile_reg", "regression")