Add feature name to model fit warnings (#31)

Concatenate the name of the feature to the warning and add a line in the log file.
biobakery · May 11, 2023 · 4bdd87d · 4bdd87d
1 parent 2a27a74
commit 4bdd87d
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/R/fit.R b/R/fit.R
@@ -232,7 +232,14 @@ fit.data <-
                             formula, 
                             data = dat_sub, 
                             na.action = na.exclude)
-                }, error = function(err) {
+                }, warning = function(w) { 
+                  message(paste("Feature", colnames(features)[x], ":", w))
+                  logging::logwarn(paste(
+                    "Fitting problem for feature", 
+                    x, 
+                    "a warning was issued"))
+                  return(fit1)
+                  }, error = function(err) {
                     fit1 <-
                         try({
                             model_function(