Add format for null data frame (#32)

boyiguo1 · Feb 18, 2022 · 466d129 · 466d129
1 parent 0184355
commit 466d129
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/R/variable_selection.R b/R/variable_selection.R
@@ -4,7 +4,10 @@ generate_var_table <- function(var_vec){
  if(length(var_vec) == 0)
  return(list(
  `Parametric` = vector(),
- `Non-parametric` = data.frame()
+ `Non-parametric` = data.frame(
+ Variable = character(),
+ Linear = logical(),
+ Nonlinear = logical())
  )
  )
 
@@ -43,7 +46,10 @@ generate_var_table <- function(var_vec){
  # Analytically Remove Functions without Linear components
  #filter(.data$Linear == TRUE)
  } else {
- non_par_df <- data.frame()
+ non_par_df <- data.frame(
+ Variable = character(),
+ Linear = logical(),
+ Nonlinear = logical())
  }
 
  return(