Use upstream URI in distributed quantile tests. (#7129)

* Use upstream URI in distributed quantile tests. * Fix test cv `PytestAssertRewriteWarning`.
2021-07-27 14:09:49 +08:00
parent e88ac9cc54
commit 7ee7a95b84
2 changed files with 12 additions and 4 deletions
--- a/tests/python/test_ranking.py
+++ b/tests/python/test_ranking.py
@@ -136,9 +136,9 @@ class TestRanking:
        cv = xgboost.cv(self.params, self.dtrain, num_boost_round=2500,
                        early_stopping_rounds=10, nfold=10, as_pandas=False)
        assert isinstance(cv, dict)
-        assert (set(cv.keys()) == {'test-ndcg-mean', 'train-ndcg-mean', 'test-ndcg-std',
-                                   'train-ndcg-std'},
-                'CV results dict key mismatch.')
+        assert set(cv.keys()) == {
+            'test-ndcg-mean', 'train-ndcg-mean', 'test-ndcg-std', 'train-ndcg-std'
+        }, "CV results dict key mismatch."

    def test_cv_no_shuffle(self):
        """