Added choosing version based on python

Signed-off-by: Elena Khaustova <[email protected]>
kedro-org · Aug 8, 2024 · e8d3c24 · e8d3c24
1 parent dc5f497
commit e8d3c24
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.5.1
+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.5.1
+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.5.1
+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.5.1
+scikit-learn~=1.5.1; python_version >= "3.9"
+scikit-learn<=1.4.0,>=1.0; python_version < "3.9"