Account for partial matching of predictor variables (#41)

Author: pat-s

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,