Fix a unit test on CLI, to handle RC versions

2020-08-02 08:56:15 +00:00 · 2020-08-02 08:56:15 +00:00 · 9116a0ec10
commit 9116a0ec10
parent 71197d1dfa
1 changed files with 2 additions and 0 deletions
--- a/tests/python/test_cli.py
+++ b/tests/python/test_cli.py
@ -121,6 +121,8 @@ eval[test] = {data_path}
        v = xgboost.__version__
        if v.find('SNAPSHOT') != -1:
            assert msg.split(':')[1].strip() == v.split('-')[0]
+        elif v.find('rc') != -1:
+            assert msg.split(':')[1].strip() == v.split('rc')[0]
        else:
            assert msg.split(':')[1].strip() == v