From 9be253632369910d2d29e006d5645d8bd6a8ec1e Mon Sep 17 00:00:00 2001
From: Rinchin <rinchin.damdinov@yandex.ru>
Date: Wed, 25 Dec 2024 15:17:55 +0000
Subject: [PATCH] scikit-learn >=0.22

---
 .github/workflows/publish_pypi.yml         | 24 ----------------------
 examples/tutorials/Tutorial_1_basics.ipynb |  2 +-
 pyproject.toml                             |  7 ++-----
 tests/integration/test_demo13.py           |  2 +-
 4 files changed, 4 insertions(+), 31 deletions(-)
 delete mode 100644 .github/workflows/publish_pypi.yml

diff --git a/.github/workflows/publish_pypi.yml b/.github/workflows/publish_pypi.yml
deleted file mode 100644
index 3cb7d8cd..00000000
--- a/.github/workflows/publish_pypi.yml
+++ /dev/null
@@ -1,24 +0,0 @@
-name: build and publish to pypi
-
-on:
-  push:
-    tags:
-        - 'v*.*.*'
-#   workflow_dispatch:
-#     inputs:
-#       tag:
-#         description: 'Tag'
-#         required: true
-#         default: 'v0.0.0'
-
-jobs:
-  deploy:
-
-    runs-on: ubuntu-latest
-
-    steps:
-    - uses: actions/checkout@v4
-
-    - uses: JRubics/poetry-publish@v2.0
-      with:
-        pypi_token: ${{ secrets.LAMA_PYPI_TOKEN }}
diff --git a/examples/tutorials/Tutorial_1_basics.ipynb b/examples/tutorials/Tutorial_1_basics.ipynb
index 4fdfe827..3854574c 100644
--- 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
index 61e4bcbf..65fb009c 100644
--- 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
index f5ac2d0a..f9223bb8 100644
--- 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