scikit-learn >=0.22

sb-ai-lab · Dec 25, 2024 · 9be2536 · 9be2536
1 parent f072c13
commit 9be2536
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 31 deletions.
diff --git a/.github/workflows/publish_pypi.yml b/.github/workflows/publish_pypi.yml
diff --git a/examples/tutorials/Tutorial_1_basics.ipynb b/examples/tutorials/Tutorial_1_basics.ipynb
@@ -1719,7 +1719,7 @@
    ],
    "source": [
     "explainer = SSWARM(automl_rd.model)\n",
-    "shap_values = explainer.shap_values(test_data, n_jobs=N_THREADS)"
+    "shap_values = explainer.shap_values(test_data)"
    ]
   },
   {

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "LightAutoML"
-version = "0.4.0"
+version = "0.4.1dev1"
 description = "Fast and customizable framework for automatic ML model creation (AutoML)"
 authors = [
     "Alexander Ryzhkov <alexmryzhkov@gmail.com>",
@@ -43,10 +43,7 @@ scipy = [
     {version = "*", python = ">=3.9"},
     {version = "<=1.11", python = "<3.9"},
 ]
-scikit-learn = [
-    {version = ">=0.22, <=0.24.2", python = "<3.10"},
-    {version = ">=0.22", python = ">=3.10"},
-]
+scikit-learn = ">=0.22"
 lightgbm = [
     {version = ">=2.3"},
     {version = "^4.4.0", platform = "darwin"}, # https://github.com/microsoft/LightGBM/issues/5328#issuecomment-2169128216

diff --git a/tests/integration/test_demo13.py b/tests/integration/test_demo13.py
@@ -46,4 +46,4 @@ def test_autots(ai92_value_77_train_test):
     forecast, _ = automl.predict(train)
 
     test_score = mean_absolute_error(test[roles["target"]].values, forecast)
-    assert test_score < 2e5
+    assert test_score < 210_00