From 5188e275135306652fb0fdb2b3a6a771c147ad8e Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Wed, 16 Aug 2023 22:44:58 +0800 Subject: [PATCH] Fix version parsing with rc release. (#9493) --- python-package/xgboost/core.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python-package/xgboost/core.py b/python-package/xgboost/core.py index fbedfd7fb..27d62cdf2 100644 --- a/python-package/xgboost/core.py +++ b/python-package/xgboost/core.py @@ -235,8 +235,11 @@ Error message(s): {os_error_list} def parse(ver: str) -> Tuple[int, int, int]: """Avoid dependency on packaging (PEP 440).""" - # 2.0.0-dev or 2.0.0 + # 2.0.0-dev, 2.0.0, or 2.0.0rc1 major, minor, patch = ver.split("-")[0].split(".") + rc = patch.find("rc") + if rc != -1: + patch = patch[:rc] return int(major), int(minor), int(patch) libver = _lib_version(lib)