Skip to content

Commit

Permalink
[Requirements] Align scikit-learn version [1.6.x] (#5627)
Browse files Browse the repository at this point in the history
  • Loading branch information
gtopper committed May 27, 2024
1 parent bdbf64c commit e5b8051
Show file tree
Hide file tree
Showing 8 changed files with 11 additions and 12 deletions.
2 changes: 1 addition & 1 deletion dev-requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,6 @@ avro~=1.11
sqlalchemy-utils~=0.39.0

# frameworks tests
scikit-learn~=1.4
scikit-learn~=1.4.0
lightgbm~=3.0; platform_machine != 'arm64'
xgboost~=1.1
2 changes: 1 addition & 1 deletion dockerfiles/base/mlrun_requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,6 @@
# TODO: delete me once we delete models entirely
matplotlib~=3.5
scipy~=1.11
scikit-learn~=1.4
scikit-learn~=1.4.0
seaborn~=0.11.0
scikit-plot~=0.3.7
2 changes: 1 addition & 1 deletion dockerfiles/jupyter/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
matplotlib~=3.5
scipy~=1.11
scikit-learn~=1.4
scikit-learn~=1.4.0
seaborn~=0.11.0
scikit-plot~=0.3.7
xgboost~=1.1
Expand Down
2 changes: 1 addition & 1 deletion dockerfiles/mlrun/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
matplotlib~=3.5
scipy~=1.11
scikit-learn~=1.4
scikit-learn~=1.4.0
seaborn~=0.11.0
scikit-plot~=0.3.7
mpi4py~=3.1
Expand Down
4 changes: 2 additions & 2 deletions docs/tutorials/01-mlrun-basics.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@
"\n",
"**Before you start, make sure the MLRun client package is installed and configured properly:**\n",
"\n",
"This notebook uses sklearn and numpy. If it is not installed in your environment run `!pip install scikit-learn~=1.4 numpy~=1.26`."
"This notebook uses sklearn and numpy. If it is not installed in your environment run `!pip install scikit-learn~=1.4.0 numpy~=1.26`."
]
},
{
Expand All @@ -82,7 +82,7 @@
"outputs": [],
"source": [
"# Install MLRun and sklearn, run this only once (restart the notebook after the install !!!)\n",
"%pip install mlrun scikit-learn~=1.4 numpy~=1.26"
"%pip install mlrun scikit-learn~=1.4.0 numpy~=1.26"
]
},
{
Expand Down
2 changes: 1 addition & 1 deletion docs/tutorials/02-model-training.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
"\n",
"## MLRun installation and configuration\n",
"\n",
"Before running this notebook make sure `mlrun` and `sklearn` packages are installed (`pip install mlrun scikit-learn~=1.4`) and that you have configured the access to the MLRun service. "
"Before running this notebook make sure `mlrun` and `sklearn` packages are installed (`pip install mlrun scikit-learn~=1.4.0`) and that you have configured the access to the MLRun service. "
]
},
{
Expand Down
8 changes: 3 additions & 5 deletions docs/tutorials/colab/01-mlrun-basics-colab.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@
"\n",
"**Before you start, make sure the MLRun client package is installed and configured properly !**\n",
"\n",
"This notebook uses sklearn. If it is not installed in your environment run `!pip install scikit-learn~=1.4`."
"This notebook uses sklearn. If it is not installed in your environment run `!pip install scikit-learn~=1.4.0`."
]
},
{
Expand All @@ -61,7 +61,7 @@
"outputs": [],
"source": [
"# install MLRun and sklearn, run this only once (restart the notebook after the install !!!)\n",
"%pip install mlrun scikit-learn~=1.4"
"%pip install mlrun scikit-learn~=1.4.0"
]
},
{
Expand Down Expand Up @@ -120,9 +120,7 @@
]
},
"outputs": [],
"source": [
"%pip install uvicorn~=0.17.0 dask-kubernetes~=0.11.0 apscheduler~=3.6 sqlite3-to-mysql~=1.4 scikit-learn~=1.4"
]
"source": "%pip install uvicorn~=0.17.0 dask-kubernetes~=0.11.0 apscheduler~=3.6 sqlite3-to-mysql~=1.4 scikit-learn~=1.4.0"
},
{
"cell_type": "markdown",
Expand Down
1 change: 1 addition & 0 deletions tests/test_requirements.py
Original file line number Diff line number Diff line change
Expand Up @@ -144,6 +144,7 @@ def test_requirement_specifiers_convention():
# conda requirements since conda does not support ~= operator
"lightgbm": {">=3.0"},
"protobuf": {"~=3.20.3", ">=3.20.3, <4"},
"scikit-learn": {"~=1.4.0"},
}

for (
Expand Down

0 comments on commit e5b8051

Please sign in to comment.