Updated test_config, skipping DIF if no gpu

benchopt · Jun 28, 2024 · eb89225 · eb89225
1 parent 5ec9975
commit eb89225
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test_config.py b/test_config.py
@@ -2,6 +2,8 @@
 
 import pytest  # noqa: F401
 
+from benchopt.utils.sys_info import get_cuda_version
+
 
 def check_test_solver_install(solver_class):
     """Hook called in `test_solver_install`.
@@ -10,4 +12,6 @@ def check_test_solver_install(solver_class):
     particular architecture, call pytest.xfail when
     detecting the situation.
     """
-    pass
+    if solver_class.name.lower() == "dif":
+        if get_cuda_version() is None:
+            pytest.xfail("Deep IsolationForest needs a working GPU hardware.")