Support doc link for the sklearn module. (#10287)

This commit is contained in:
Jiaming Yuan
2024-08-06 02:35:32 +08:00
committed by GitHub
parent a269055b2b
commit 3d8107adb8
8 changed files with 92 additions and 13 deletions

View File

@@ -9,6 +9,7 @@ import pytest
import xgboost as xgb
from xgboost import testing as tm
from xgboost.core import _parse_version
dpath = "demo/data/"
rng = np.random.RandomState(1994)
@@ -315,3 +316,14 @@ class TestBasicPathLike:
"""An invalid model_file path should raise XGBoostError."""
with pytest.raises(xgb.core.XGBoostError):
xgb.Booster(model_file=Path("invalidpath"))
def test_parse_ver() -> None:
(major, minor, patch), post = _parse_version("2.1.0")
assert post == ""
(major, minor, patch), post = _parse_version("2.1.0-dev")
assert post == "dev"
(major, minor, patch), post = _parse_version("2.1.0rc1")
assert post == "rc1"
(major, minor, patch), post = _parse_version("2.1.0.post1")
assert post == "post1"

View File

@@ -1484,3 +1484,16 @@ def test_tags() -> None:
tags = xgb.XGBRanker()._more_tags()
assert "multioutput" not in tags
def test_doc_link() -> None:
for est in [
xgb.XGBRegressor(),
xgb.XGBClassifier(),
xgb.XGBRanker(),
xgb.XGBRFRegressor(),
xgb.XGBRFClassifier(),
]:
name = est.__class__.__name__
link = est._get_doc_link()
assert f"xgboost.{name}" in link