different xgb versions for different pythons

sb-ai-lab · Jul 31, 2024 · 45c80ba · 45c80ba
1 parent 17c9f24
commit 45c80ba
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -44,7 +44,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-20.04, windows-latest, macos-latest]  # FIX: 'ubuntu-latest'(ubunut-22.04) -> 'ubuntu-20.04': Python version 3.6 was not found in the local cache
-        python-version: ["3.6", "3.7", "3.8", "3.9", "3.10"]
+        python-version: ["3.6", "3.7", "3.8", "3.9", "3.10", "3.11"]
 
     steps:
       - uses: actions/checkout@v2

diff --git a/pyproject.toml b/pyproject.toml
@@ -50,7 +50,10 @@ scikit-learn = [
 ]
 lightgbm = ">=2.3, <=3.2.1"
 catboost = ">=0.26.1"
-xgboost = "^2.0.0"
+xgboost = [
+    {version = "*", python = "<3.8"},
+    {version = "^2.0.0", python = ">=3.8"}
+]
 optuna = "*"
 torch = [
     {platform = "win32", python = "3.6.1", version = "1.7.0"},

diff --git a/tests/unit/test_automl/test_presets/test_whiteboxpreset.py b/tests/unit/test_automl/test_presets/test_whiteboxpreset.py
@@ -21,7 +21,7 @@ def test_fit_predict(self, jobs_train_test, jobs_roles, binary_task):
         ho_score = roc_auc_score(test[target_name].values, ho_predictions.data[:, 0])
 
         # checks
-        assert oof_score > 0.78
-        assert ho_score > 0.78
+        assert oof_score > 0.75
+        assert ho_score > 0.75
 
         check_pickling(automl, ho_score, binary_task, test, target_name)