forgot NAs are expected in smry

ropensci · Oct 7, 2023 · 9551eba · 9551eba
1 parent 386f66e
commit 9551eba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/orsf_pd.R b/R/orsf_pd.R
@@ -539,7 +539,7 @@ orsf_pred_dependence <- function(object,
  out <- rbindlist(pd_vals)
 
  # missings may occur when oobag=TRUE and n_tree is small
- out <- collapse::na_omit(out)
+ if(type_output == 'ice') out <- collapse::na_omit(out, cols = 'pred')
 
  ids <- c('id_variable', if(type_output == 'ice') 'id_row')