diff --git a/Jenkinsfile-win64 b/Jenkinsfile-win64 index 15dc345ed2ac..479a78cd94f3 100644 --- a/Jenkinsfile-win64 +++ b/Jenkinsfile-win64 @@ -28,7 +28,7 @@ pipeline { steps { script { parallel ([ - 'build-win64-cuda9.0': { BuildWin64() } + 'build-win64-cuda10.0': { BuildWin64() } ]) } milestone ordinal: 2 @@ -40,7 +40,6 @@ pipeline { script { parallel ([ 'test-win64-cpu': { TestWin64CPU() }, - 'test-win64-gpu-cuda9.0': { TestWin64GPU(cuda_target: 'cuda9') }, 'test-win64-gpu-cuda10.0': { TestWin64GPU(cuda_target: 'cuda10_0') }, 'test-win64-gpu-cuda10.1': { TestWin64GPU(cuda_target: 'cuda10_1') } ]) @@ -67,7 +66,7 @@ def checkoutSrcs() { } def BuildWin64() { - node('win64 && build') { + node('win64 && build && cuda10') { unstash name: 'srcs' echo "Building XGBoost for Windows AMD64 target..." bat "nvcc --version" diff --git a/python-package/xgboost/core.py b/python-package/xgboost/core.py index 6ea735a162a0..72ffe23662e7 100644 --- a/python-package/xgboost/core.py +++ b/python-package/xgboost/core.py @@ -1588,7 +1588,7 @@ def predict(self, ngroup = int(chunk_size / ((data.num_col() + 1) * (data.num_col() + 1))) if ngroup == 1: - preds = preds.reshape(nrow, + preds = preds.reshape(nrow, # pylint: disable=too-many-function-args data.num_col() + 1, data.num_col() + 1) else: