actually fix test this time maybe?

openvolley · Feb 16, 2024 · 1dbf719 · 1dbf719
1 parent d059751
commit 1dbf719
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/testthat/test-network.R b/tests/testthat/test-network.R
@@ -16,7 +16,11 @@ test_that("basic inference", {
 
 
 test_that("cuda fails on cpu-only system", {
-    have_cuda <- reticulate::py_run_string("import torch\ncuda=torch.cuda.is_available()")$cuda
+    have_cuda <- tryCatch({
+        reticulate::use_virtualenv(ovml_yolo7_python_envname())
+        reticulate::py_run_string("import torch\ncuda=torch.cuda.is_available()")$cuda
+    }, error = function(e) NA)
+    skip_if(is.na(have_cuda))
     if (!have_cuda) {
         expect_warning(ovml_yolo(device = 0), "device not available")
     }