Skip dask tests on ARM. (#6267)

Co-authored-by: Hyunsu Cho <chohyu01@cs.washington.edu>

tests/python/test_demos.py

@@ -39,7 +39,6 @@ def test_feature_weights_demo():
 
 
 @pytest.mark.skipif(**tm.no_sklearn())
-@pytest.mark.skipif(**tm.is_arm())
 def test_sklearn_demo():
     script = os.path.join(PYTHON_DEMO_DIR, 'sklearn_examples.py')
     cmd = ['python', script]
@@ -49,7 +48,6 @@ def test_sklearn_demo():
 
 
 @pytest.mark.skipif(**tm.no_sklearn())
-@pytest.mark.skipif(**tm.is_arm())
 def test_sklearn_parallel_demo():
     script = os.path.join(PYTHON_DEMO_DIR, 'sklearn_parallel.py')
     cmd = ['python', script]

tests/python/test_early_stopping.py

@@ -83,7 +83,6 @@ class TestEarlyStopping(unittest.TestCase):
 
     @pytest.mark.skipif(**tm.no_sklearn())
     @pytest.mark.skipif(**tm.no_pandas())
-    @pytest.mark.skipif(**tm.is_arm())
     def test_cv_early_stopping_with_multiple_eval_sets_and_metrics(self):
         from sklearn.datasets import load_breast_cancer
 

tests/python/test_with_dask.py

@@ -8,13 +8,14 @@ import asyncio
 from sklearn.datasets import make_classification
 import os
 import subprocess
-from hypothesis import given, strategies, settings, note
+from hypothesis import given, settings, note
 from test_updaters import hist_parameter_strategy, exact_parameter_strategy
 
 if sys.platform.startswith("win"):
     pytest.skip("Skipping dask tests on Windows", allow_module_level=True)
+if tm.no_dask()['condition']:
+    pytest.skip(msg=tm.no_dask()['reason'], allow_module_level=True)
 
-pytestmark = pytest.mark.skipif(**tm.no_dask())
 
 try:
     from distributed import LocalCluster, Client, get_client
@@ -636,7 +637,6 @@ class TestWithDask:
         # Make sure that it's decreasing
         assert history[-1] < history[0]
 
-    @pytest.mark.skipif(**tm.is_arm())
     @given(params=hist_parameter_strategy,
            dataset=tm.dataset_strategy)
     @settings(deadline=None)
@@ -644,7 +644,6 @@ class TestWithDask:
         num_rounds = 30
         self.run_updater_test(client, params, num_rounds, dataset, 'hist')
 
-    @pytest.mark.skipif(**tm.is_arm())
     @given(params=exact_parameter_strategy,
            dataset=tm.dataset_strategy)
     @settings(deadline=None)

tests/python/testing.py

@@ -23,10 +23,6 @@ except ImportError:
 memory = Memory('./cachedir', verbose=0)
 
 
-def is_arm():
-    return {'condition': platform.machine().lower().find('arm') != -1,
-            'reason': 'Skipping expensive tests on ARM.'}
-
 def no_sklearn():
     return {'condition': not SKLEARN_INSTALLED,
             'reason': 'Scikit-Learn is not installed'}