move the gpu dask tests running behind spark

the gpu dask tests may corrupt the whole gpu env
dmlc · Aug 2, 2022 · 0f388d9 · 0f388d9
1 parent 2096997
commit 0f388d9
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 3 deletions.
diff --git a/tests/python-gpu/conftest.py b/tests/python-gpu/conftest.py
@@ -61,8 +61,8 @@ def pytest_collection_modifyitems(config, items):
     mgpu_mark = pytest.mark.mgpu
     for item in items:
         if item.nodeid.startswith(
-            "python-gpu/test_gpu_with_dask.py"
+            "python-gpu/test_gpu_with_dask/test_gpu_with_dask.py"
         ) or item.nodeid.startswith(
-            "python-gpu/test_spark_with_gpu/test_spark_with_gpu.py"
+            "python-gpu/test_gpu_spark/test_gpu_spark.py"
         ):
             item.add_marker(mgpu_mark)
diff --git a/...n-gpu/test_spark_with_gpu/discover_gpu.sh → ...python-gpu/test_gpu_spark/discover_gpu.sh b/...n-gpu/test_spark_with_gpu/discover_gpu.sh → ...python-gpu/test_gpu_spark/discover_gpu.sh
diff --git a/...thon-gpu/test_spark_with_gpu/test_data.py → tests/python-gpu/test_gpu_spark/test_data.py b/...thon-gpu/test_spark_with_gpu/test_data.py → tests/python-gpu/test_gpu_spark/test_data.py
diff --git a/...est_spark_with_gpu/test_spark_with_gpu.py → ...thon-gpu/test_gpu_spark/test_gpu_spark.py b/...est_spark_with_gpu/test_spark_with_gpu.py → ...thon-gpu/test_gpu_spark/test_gpu_spark.py
@@ -17,7 +17,7 @@
 from pyspark.sql import SparkSession
 from xgboost.spark import SparkXGBClassifier, SparkXGBRegressor
 
-gpu_discovery_script_path = "tests/python-gpu/test_spark_with_gpu/discover_gpu.sh"
+gpu_discovery_script_path = "tests/python-gpu/test_gpu_spark/discover_gpu.sh"
 executor_gpu_amount = 4
 executor_cores = 4
 num_workers = executor_gpu_amount

diff --git a/tests/python-gpu/test_gpu_with_dask.py → .../test_gpu_with_dask/test_gpu_with_dask.py b/tests/python-gpu/test_gpu_with_dask.py → .../test_gpu_with_dask/test_gpu_with_dask.py