diff --git a/tests/python-sycl/test_sycl_training_continuation.py b/tests/python-sycl/test_sycl_training_continuation.py index e2a11c987bb4..ca4ba5233283 100644 --- a/tests/python-sycl/test_sycl_training_continuation.py +++ b/tests/python-sycl/test_sycl_training_continuation.py @@ -13,7 +13,7 @@ def run_training_continuation(self, use_json): y = np.random.randn(kRows) dtrain = xgb.DMatrix(X, y) params = { - "device": "sycl", + "device": "sycl:gpu", "max_depth": "2", "gamma": "0.1", "alpha": "0.01",