From c3ca8abc65e790ce98034d07c756258097b445eb Mon Sep 17 00:00:00 2001 From: "C. Regouby" Date: Wed, 17 Apr 2024 17:14:56 +0200 Subject: [PATCH] fix a warning in autoplot --- tests/testthat/test-plot.R | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/testthat/test-plot.R b/tests/testthat/test-plot.R index 1790024..f5bd125 100644 --- a/tests/testthat/test-plot.R +++ b/tests/testthat/test-plot.R @@ -60,8 +60,9 @@ test_that("Autoplot of pretrain then fit scenario, pretrain without checkpoints, print(autoplot(fit_no_checkpoint)) ) fit_with_checkpoint <- tabnet_fit(Sale_Price ~., data = small_ames, tabnet_model = fit_no_checkpoint, epochs = 2, checkpoint_epoch = 1) - expect_no_error( - print(autoplot(fit_with_checkpoint)) + expect_warning( + print(autoplot(fit_with_checkpoint)), + "Removed 2 rows containing missing values" ) })