[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>
dmlc · May 31, 2020 · e74560c · e74560c
1 parent 882b966
commit e74560c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git 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
@@ -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: