Added exception weibull regression for training.

alexzwanenburg · Feb 3, 2025 · 613eadf · 613eadf
1 parent e4a3a5f
commit 613eadf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/R/TestFunctions.R b/R/TestFunctions.R
@@ -44,6 +44,7 @@ test_all_learners_train_predict_vimp <- function(
     learners,
     hyperparameter_list = NULL,
     except_train = NULL,
+    except_train_partial_prospective = NULL,
     except_naive = NULL,
     except_predict = NULL,
     except_predict_survival = NULL,
@@ -1423,7 +1424,7 @@ test_all_learners_train_predict_vimp <- function(
         ),
         {
           # Test that the model was successfully created.
-          testthat::expect_equal(model_is_trained(model), !learner %in% except_train)
+          testthat::expect_equal(model_is_trained(model), !learner %in% c(except_train, except_train_partial_prospective))
         }
       )
     }

diff --git a/tests/testthat/test-learner_survival_regression_S4.R b/tests/testthat/test-learner_survival_regression_S4.R
@@ -9,7 +9,8 @@ testthat::skip_on_cran()
 testthat::skip_on_ci()
 
 familiar:::test_all_learners_train_predict_vimp(
-  learners = familiar:::.get_available_survival_regression_learners(show_general = FALSE)
+  learners = familiar:::.get_available_survival_regression_learners(show_general = FALSE),
+  except_train_partial_prospective = "survival_regr_weibull"
 )
 familiar:::test_all_learners_parallel_train_predict_vimp(
   learners = familiar:::.get_available_survival_regression_learners(show_general = FALSE)