Skip to content

Commit

Permalink
dont use mlr3 dev features
Browse files Browse the repository at this point in the history
  • Loading branch information
sebffischer committed Feb 7, 2025
1 parent a5e7403 commit b45600d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/testthat/test_LearnerTorchModel.R
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ test_that("LearnerTorchModel and marshaling", {
packages = "data.table",
)
learner$encapsulate("callr", lrn("classif.featureless"))
learner$configure(
learner$param_set$set_values(
batch_size = 50,
epochs = 1
)
Expand Down

0 comments on commit b45600d

Please sign in to comment.