...

mlr-org · Oct 16, 2024 · b039531 · b039531
1 parent 30c65ab
commit b039531
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/testthat/test_with_torch_settings.R b/tests/testthat/test_with_torch_settings.R
@@ -45,10 +45,11 @@ test_that("interop threads work", {
     library(torch)
     with_torch_settings = getFromNamespace("with_torch_settings", "mlr3torch")
     with_torch_settings(NULL, 1, 2, invisible(NULL))
-    x1 = capture.output(with_torch_settings(NULL, 1, 2, invisible(NULL)))
-    x2 = capture.output(with_torch_settings(NULL, 1, 1, invisible(NULL)))
-    list(x1, x2)
+    x1 = tryCatch(with_torch_settings(NULL, 1, 2, invisible(NULL)), warning = identity)$message
+    x2 = tryCatch(with_torch_settings(NULL, 1, 1, invisible(NULL)), warning = identity)$message
+    list(x1, x2, torch_get_num_interop_threads())
   })
-  expect_equal(result[[1]], character(0))
+  expect_true(length(result[[1]]) == 0)
   expect_true(grepl("keeping the previous value 2", result[[2]], fixed = TRUE))
+  expect_equal(result[[3]], 2)
 })