mark mgpu test.

dmlc · Mar 16, 2021 · 808c33a · 808c33a
1 parent 9b24ba8
commit 808c33a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/python-gpu/conftest.py b/tests/python-gpu/conftest.py
@@ -53,3 +53,9 @@ def pytest_collection_modifyitems(config, items):
         for item in items:
             if any(item.nodeid.startswith(x) for x in blocklist):
                 item.add_marker(skip_mark)
+
+    mgpu_mark = pytest.mark.mgpu
+    for item in items:
+        if item.nodeid.startswith("python-gpu/test_gpu_with_dask.py"):
+            print("mark:", item, item.nodeid)
+            item.add_marker(mgpu_mark)