Run actions again

jinlow · May 9, 2024 · 1487b66 · 1487b66
1 parent ea03858
commit 1487b66
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -67,9 +67,7 @@ jobs:
         uses: dtolnay/rust-toolchain@stable
         with:
           toolchain: stable
-      - uses: actions/setup-python@v5
-        env:
-            PIP_DISABLE_PIP_VERSION_CHECK: 1
+      - uses: actions/setup-python@v4
         with:
           python-version: ${{ matrix.pyversion }}
           architecture: x64

diff --git a/py-forust/tests/test_booster.py b/py-forust/tests/test_booster.py
@@ -1156,6 +1156,7 @@ def test_booster_to_xgboosts_with_contributions_shapley_from_xgboost(X_y):
     assert np.allclose(contribs_shapley, xmod_contribs_shapley, atol=0.00001)
     assert np.allclose(fmod_preds, xmod.predict(X, output_margin=True), atol=0.00001)
 
+
 @pytest.mark.parametrize("growth_policy", ["depthwise", "lossguide"])
 def test_from_xgboost(X_y, growth_policy):
     X, y = X_y
@@ -1173,7 +1174,10 @@ def test_from_xgboost(X_y, growth_policy):
     xmod.fit(X, y)
 
     fmod = forust._from_xgboost_model(xmod)
-    assert np.allclose(xmod.predict(X, output_margin=True), fmod.predict(X), atol=0.00001)
+    assert np.allclose(
+        xmod.predict(X, output_margin=True), fmod.predict(X), atol=0.00001
+    )
+
 
 def test_missing_branch_with_contributions(X_y):
     X, y = X_y