From e74560c86a8dec01ba93a5ce327b4c8df1c25fc2 Mon Sep 17 00:00:00 2001
From: Jiaming Yuan <jm.yuan@outlook.com>
Date: Sun, 31 May 2020 21:35:00 +0800
Subject: [PATCH] [CI] Backport Remove CUDA 9.0 from Windows CI. (#5674)
 (#5738)

* [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 <chohyu01@cs.washington.edu>

* Pylint.

Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
---
 Jenkinsfile-win64              | 5 ++---
 python-package/xgboost/core.py | 2 +-
 2 files changed, 3 insertions(+), 4 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"
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: