Merge branch 'dev' into ci/requirements

awslabs · Feb 2, 2024 · 39d5029 · 39d5029
2 parents bf3e0e9 + 9ea944b
commit 39d5029
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/gluonts/nursery/daf/requirements.txt b/src/gluonts/nursery/daf/requirements.txt
@@ -2,6 +2,6 @@ torch==1.13.1
 tensorboard==2.3.0
 numpy==1.22.0
 pandas==1.1.5
-scikit-learn==1.0.1
+scikit-learn~=1.4.0
 scipy==1.10.0
 matplotlib==3.3.2
diff --git a/src/gluonts/nursery/few_shot_prediction/pyproject.toml b/src/gluonts/nursery/few_shot_prediction/pyproject.toml
@@ -17,7 +17,7 @@ pandas = "^1.3.1"
 python = "^3.8,<3.10"
 pytorch-lightning = "^1.4.4"
 sagemaker = "^2.40.0,<2.41.0"
-scikit-learn = "^0.24.2"
+scikit-learn = "^1.4.0"
 torch = "^1.9.0"
 sagemaker-training = "^3.9.2"
 python-dotenv = "^0.19.0"

diff --git a/src/gluonts/nursery/tsbench/pyproject.toml b/src/gluonts/nursery/tsbench/pyproject.toml
@@ -32,7 +32,7 @@ pytorch-lightning = "^1.5.0"
 rpy2 = ">=2.9.*,<3.*"
 sagemaker = "^2.40.0"
 sagemaker-training = "^3.9.2"
-scikit-learn = "^0.24.2"
+scikit-learn = "^1.4.0"
 scipy = "^1.6.3"
 seaborn = "^0.11.2"
 statsmodels = "^0.13.0"