Update custom model to use new_model_spec()

tidymodels · Apr 3, 2024 · 337033e · 337033e
1 parent 016a5ce
commit 337033e
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/tests/testthat/test-mda.R b/tests/testthat/test-mda.R
@@ -46,13 +46,14 @@ test_that("mda + custom parsnip model + predict() works", {
       stop("`mode` should be 'classification'", call. = FALSE)
     }
     args <- list(sub_classes = rlang::enquo(sub_classes))
-    out <- list(args = args,
-                eng_args = NULL,
-                mode = mode,
-                method = NULL,
-                engine = NULL)
-    class(out) <- make_classes("mixture_da")
-    out
+    new_model_spec(
+      "mixture_da",
+      args = args,
+      eng_args = NULL,
+      mode = mode,
+      method = NULL,
+      engine = NULL
+    )
   }
   set_fit(
     model = "mixture_da",