Account for partial matching of predictor variables (#41)

pat-s · May 24, 2020 · 590928f · 590928f
1 parent f5f2388
commit 590928f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/helper_funs.R b/R/helper_funs.R
@@ -74,7 +74,7 @@ gam_to_df <- function(model = NULL, pred = NULL) {
   plot_df <- no_plot(model) # nolint
 
   # get list index of spec. predictor
-  set_pred <- which(grepl(pred, plot_df))
+  set_pred <- grep(paste0("\\b", pred, "\\b"), plot_df)
 
   df <- data.frame(
     x = plot_df[[set_pred]]$x,