Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CI] Backport Remove CUDA 9.0 from Windows CI. (#5674) (#5738)

Browse Source 
* [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>
This commit is contained in:
Jiaming Yuan
2020-05-31 21:35:00 +08:00
committed by GitHub
parent 882b966536
commit e74560c86a
2 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python-package/xgboost/core.py
View File

@@ -1588,7 +1588,7 @@ class Booster(object):
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: