Merge branch 'main' into no-pkg

NEWS.md

@@ -35,6 +35,8 @@
 
 * Fixed bug related to using local (non-package) models (#1229)
 
+* `tunable()` now references a dials object for the `mixture` parameter (#1236)
+
 ## Breaking Change
 
 * For quantile prediction, the `quantile` argument to `predict()` has been deprecate in facor of `quantile_levels`. This does not affect models with mode `"quantile regression"`. 

R/tunable.R

@@ -206,10 +206,9 @@ brulee_mlp_engine_args <-
     "largest",                         list(pkg = "dials", fun = "rate_largest"),
     "rate_schedule",                  list(pkg = "dials", fun = "rate_schedule"),
     "step_size",                     list(pkg = "dials", fun = "rate_step_size"),
-    "steps",                             list(pkg = "dials", fun = "rate_steps")
+    "mixture",                              list(pkg = "dials", fun = "mixture")
   ) %>%
-  dplyr::mutate(,
-                source = "model_spec",
+  dplyr::mutate(source = "model_spec",
                 component = "mlp",
                 component_id = "engine"
   )

parsnip.Rproj

@@ -1,5 +1,6 @@
 Version: 1.0
-ProjectId: e5169c4e-5aba-443d-938b-8765efc1d040
+ProjectId: 7f6c9ff5-6b9a-4235-8666-12db5ef65d49
+
 
 RestoreWorkspace: No
 SaveWorkspace: No

tests/testthat/test-tunable.R

@@ -0,0 +1,22 @@
+test_that('brulee has mixture object', {
+  # for issue 1236
+  mlp_spec <-
+    mlp(
+      hidden_units = tune(),
+      activation =  tune(),
+      penalty = tune(),
+      learn_rate = tune(),
+      epoch = 2000
+    ) %>%
+    set_mode("regression") %>%
+    set_engine("brulee",
+               stop_iter = tune(),
+               mixture = tune(),
+               rate_schedule = tune())
+
+  brulee_res <- tunable(mlp_spec)
+
+  expect_true(
+    length(brulee_res$call_info[brulee_res$name == "mixture"]) > 0
+  )
+})