From 93f3acdef920a149a7fd42fe9f6bf16871b84032 Mon Sep 17 00:00:00 2001
From: Jiaming Yuan <jm.yuan@outlook.com>
Date: Fri, 2 Jul 2021 21:26:00 +0800
Subject: [PATCH] Fix with latest pylint. (#7071)

---
 python-package/setup.py            | 12 ++++++++----
 python-package/xgboost/training.py |  2 +-
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/python-package/setup.py b/python-package/setup.py
index a2feb7667..b77e27b0c 100644
--- a/python-package/setup.py
+++ b/python-package/setup.py
@@ -301,12 +301,16 @@ if __name__ == '__main__':
     # - pip install . -e
     # - python setup.py develop   # same as above
     logging.basicConfig(level=logging.INFO)
+
+    with open(os.path.join(CURRENT_DIR, 'README.rst'), encoding='utf-8') as fd:
+        description = fd.read()
+    with open(os.path.join(CURRENT_DIR, 'xgboost/VERSION')) as fd:
+        version = fd.read().strip()
+
     setup(name='xgboost',
-          version=open(os.path.join(
-              CURRENT_DIR, 'xgboost/VERSION')).read().strip(),
+          version=version,
           description="XGBoost Python Package",
-          long_description=open(os.path.join(CURRENT_DIR, 'README.rst'),
-                                encoding='utf-8').read(),
+          long_description=description,
           long_description_content_type="text/x-rst",
           install_requires=[
               'numpy',
diff --git a/python-package/xgboost/training.py b/python-package/xgboost/training.py
index 3bbfc1291..1fe298362 100644
--- a/python-package/xgboost/training.py
+++ b/python-package/xgboost/training.py
@@ -502,7 +502,7 @@ def cv(params, dtrain, num_boost_round=10, nfold=3, stratified=False, folds=None
             results[key + '-std'].append(std)
 
         if should_break:
-            for k in results:
+            for k in results.keys():  # pylint: disable=consider-iterating-dictionary
                 results[k] = results[k][:(booster.best_iteration + 1)]
             break
     if as_pandas: