| FazBrowse GitHub Viewer | Trending | | Home |
| Tools: [Download Repo ZIP] [Original HTTPS Page] |
1 file changed
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -1,9 +1,16 @@ | |||
| 1 | 1 | import unittest | |
| 2 | 2 | ||
| 3 | + import xgboost | ||
| 4 | + | ||
| 5 | + from distutils.version import StrictVersion | ||
| 3 | 6 | from sklearn import datasets | |
| 4 | 7 | from xgboost import XGBClassifier | |
| 5 | 8 | ||
| 6 | 9 | class TestXGBoost(unittest.TestCase): | |
| 10 | + def test_version(self): | ||
| 11 | + # b/175051617 prevent xgboost version downgrade. | ||
| 12 | + self.assertGreaterEqual(StrictVersion(xgboost.__version__), StrictVersion("1.2.1")) | ||
| 13 | + | ||
| 7 | 14 | def test_classifier(self): | |
| 8 | 15 | boston = datasets.load_boston() | |
| 9 | 16 | X, y = boston.data, boston.target | |
| Back | FazBrowse Home | New Git URL |
0 commit comments