Updated scikit-learn version (#233)

* Updated scikit-learn version Signed-off-by: Elena Khaustova <[email protected]> * Added choosing version based on python Signed-off-by: Elena Khaustova <[email protected]> --------- Signed-off-by: Elena Khaustova <[email protected]>
kedro-org · Aug 12, 2024 · 4602050 · 4602050
1 parent 75908ef
commit 4602050
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/spaceflights-pandas-viz/{{ cookiecutter.repo_name }}/requirements.txt b/spaceflights-pandas-viz/{{ cookiecutter.repo_name }}/requirements.txt
@@ -9,5 +9,6 @@ pytest-cov~=3.0
 pytest-mock>=1.7.1, <2.0
 pytest~=7.2
 ruff~=0.1.8
-scikit-learn~=1.0
+scikit-learn~=1.5.1; python_version >= "3.9"
+scikit-learn<=1.4.0,>=1.0; python_version < "3.9"
 seaborn~=0.12.1
diff --git a/spaceflights-pandas/{{ cookiecutter.repo_name }}/requirements.txt b/spaceflights-pandas/{{ cookiecutter.repo_name }}/requirements.txt
@@ -9,4 +9,5 @@ pytest-cov~=3.0
 pytest-mock>=1.7.1, <2.0
 pytest~=7.2
 ruff~=0.1.8
-scikit-learn~=1.0
+scikit-learn~=1.5.1; python_version >= "3.9"
+scikit-learn<=1.4.0,>=1.0; python_version < "3.9"
diff --git a/spaceflights-pyspark-viz/{{ cookiecutter.repo_name }}/requirements.txt b/spaceflights-pyspark-viz/{{ cookiecutter.repo_name }}/requirements.txt
@@ -9,6 +9,7 @@ pytest-cov~=3.0
 pytest-mock>=1.7.1, <2.0
 pytest~=7.2
 ruff~=0.1.8
-scikit-learn~=1.0
+scikit-learn~=1.5.1; python_version >= "3.9"
+scikit-learn<=1.4.0,>=1.0; python_version < "3.9"
 seaborn~=0.12.1
 setuptools; python_version >= "3.12"
diff --git a/spaceflights-pyspark/{{ cookiecutter.repo_name }}/requirements.txt b/spaceflights-pyspark/{{ cookiecutter.repo_name }}/requirements.txt
@@ -9,4 +9,5 @@ pytest-cov~=3.0
 pytest-mock>=1.7.1, <2.0
 pytest~=7.2
 ruff~=0.1.8
-scikit-learn~=1.0
+scikit-learn~=1.5.1; python_version >= "3.9"
+scikit-learn<=1.4.0,>=1.0; python_version < "3.9"