Fix CI with latest xgboost (#257)

* Fix CI with latest xgboost

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>

* trigger CI

* Update github actions to v3

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>

Signed-off-by: Antoni Baum <antoni.baum@protonmail.com>

Author: Yard1

.github/workflows/gpu.yaml

@@ -8,9 +8,9 @@ jobs:
     runs-on: ubuntu-latest
     timeout-minutes: 20
     steps:
-    - uses: actions/checkout@v2
+    - uses: actions/checkout@v3
     - name: Set up Python 3.7
-      uses: actions/setup-python@v2
+      uses: actions/setup-python@v3
       with:
         python-version: 3.7
     - name: Install dependencies

.github/workflows/test.yaml

@@ -11,9 +11,9 @@ jobs:
     runs-on: ubuntu-latest
     timeout-minutes: 3
     steps:
-    - uses: actions/checkout@v2
+    - uses: actions/checkout@v3
     - name: Set up Python 3.7
-      uses: actions/setup-python@v2
+      uses: actions/setup-python@v3
       with:
         python-version: 3.7
     - name: Install dependencies
@@ -42,9 +42,9 @@ jobs:
           - python-version: 3.9
             ray-wheel: https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-3.0.0.dev0-cp39-cp39-manylinux2014_x86_64.whl
     steps:
-    - uses: actions/checkout@v2
+    - uses: actions/checkout@v3
     - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v2
+      uses: actions/setup-python@v3
       with:
         python-version: ${{ matrix.python-version }}
     - name: Install dependencies
@@ -79,9 +79,9 @@ jobs:
       matrix:
         python-version: [3.7, 3.8, 3.9]
     steps:
-    - uses: actions/checkout@v2
+    - uses: actions/checkout@v3
     - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v2
+      uses: actions/setup-python@v3
       with:
         python-version: ${{ matrix.python-version }}
     - name: Install dependencies
@@ -118,9 +118,9 @@ jobs:
       matrix:
         python-version: [3.7, 3.8, 3.9]
     steps:
-    - uses: actions/checkout@v2
+    - uses: actions/checkout@v3
     - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v2
+      uses: actions/setup-python@v3
       with:
         python-version: ${{ matrix.python-version }}
     - name: Install dependencies
@@ -169,9 +169,9 @@ jobs:
           - python-version: 3.9
             ray-wheel: https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-3.0.0.dev0-cp39-cp39-manylinux2014_x86_64.whl
     steps:
-    - uses: actions/checkout@v2
+    - uses: actions/checkout@v3
     - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v2
+      uses: actions/setup-python@v3
       with:
         python-version: ${{ matrix.python-version }}
     - name: Install dependencies
@@ -187,7 +187,7 @@ jobs:
       run: |
         python -m pip install -e .
     - name: Clone XGBoost repo
-      uses: actions/checkout@v2
+      uses: actions/checkout@v3
       with:
         repository: dmlc/xgboost
         path: xgboost
@@ -224,9 +224,9 @@ jobs:
       matrix:
         python-version: [3.7]
     steps:
-    - uses: actions/checkout@v2
+    - uses: actions/checkout@v3
     - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v2
+      uses: actions/setup-python@v3
       with:
         python-version: ${{ matrix.python-version }}
     - name: Install dependencies

xgboost_ray/tests/test_sklearn.py

@@ -76,6 +76,14 @@ def objective(labels, predt):
     return objective
 
 
+def get_basescore(model: xgb.XGBModel) -> float:
+    """Get base score from an XGBoost sklearn estimator."""
+    base_score = float(
+        json.loads(model.get_booster().save_config())["learner"][
+            "learner_model_param"]["base_score"])
+    return base_score
+
+
 class TemporaryDirectory(object):
     """Context manager for tempfile.mkdtemp()"""
 
@@ -628,7 +636,8 @@ def test_sklearn_get_default_params(self):
         cls = RayXGBClassifier()
         assert cls.get_params()["base_score"] is None
         cls.fit(X[:4, ...], y[:4, ...])
-        assert cls.get_params()["base_score"] is not None
+        base_score = get_basescore(cls)
+        np.testing.assert_equal(base_score, 0.5)
 
     @unittest.skipIf(XGBOOST_VERSION < Version("1.1.0"),
                      f"not supported in xgb version {xgb.__version__}")