align test with improved message

mlverse · Sep 26, 2023 · 209d04b · 209d04b
1 parent 859d9dc
commit 209d04b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/testthat/test-hardhat_parameters.R b/tests/testthat/test-hardhat_parameters.R
@@ -162,7 +162,7 @@ test_that("fit raise an error with non-supported mask-type", {
   expect_error(
     tabnet_fit(rec, attrition, epochs = 1, valid_split = 0.25, verbose = TRUE,
                       config = tabnet_config( mask_type="max_entropy")),
-    regexp = "either sparsemax or entmax"
+    regexp = "either 'sparsemax' or 'entmax' as"
   )
 
 })