From e6f08349e8dfe4a793f27bbd41bdbd3b0027f62c Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Thu, 28 May 2020 03:23:36 +0800 Subject: [PATCH 1/2] [CI] Remove CUDA 9.0 from Windows CI. (#5674) * Remove CUDA 9.0 on Windows CI. * Require cuda10 tag, to differentiate Co-authored-by: Philip Hyunsu Cho --- Jenkinsfile-win64 | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) 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" From 2986c4b130cf0b9e8f8bbc8ab2b12990b6933169 Mon Sep 17 00:00:00 2001 From: fis Date: Sun, 31 May 2020 18:57:41 +0800 Subject: [PATCH 2/2] Pylint. --- python-package/xgboost/core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: