Update test_dask.py

microsoft · Feb 12, 2021 · fad0b69 · fad0b69
1 parent 3c0b6c4
commit fad0b69
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/python_package_test/test_dask.py b/tests/python_package_test/test_dask.py
@@ -5,6 +5,7 @@
 import pickle
 import socket
 from itertools import groupby
+from platform import machine
 from os import getenv
 from sys import platform
 
@@ -42,7 +43,8 @@
 
 pytestmark = [
     pytest.mark.skipif(getenv('TASK', '') == 'mpi', reason='Fails to run with MPI interface'),
-    pytest.mark.skipif(getenv('TASK', '') == 'gpu', reason='Fails to run with GPU interface')
+    pytest.mark.skipif(getenv('TASK', '') == 'gpu', reason='Fails to run with GPU interface'),
+    pytest.mark.skipif(machine() != 'x86_64', reason='Fails to run with non-x86_64 architecture')
 ]