From 7668ce11fc296de18daf71d718f91b2f447e3615 Mon Sep 17 00:00:00 2001
From: Lorenzo Stella <stellalo@amazon.com>
Date: Thu, 25 Jul 2024 14:17:22 +0200
Subject: [PATCH] Upgrade dependencies in nursery (#3206)

*Issue #, if available:*
- https://github.com/awslabs/gluonts/security/dependabot/24
- https://github.com/awslabs/gluonts/security/dependabot/27
- https://github.com/awslabs/gluonts/security/dependabot/28
- https://github.com/awslabs/gluonts/security/dependabot/29
- https://github.com/awslabs/gluonts/security/dependabot/30

*Description of changes:* Upgrade dependencies to recommended versions.


By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.


**Please tag this pr with at least one of these labels to make our
release process faster:** BREAKING, new feature, bug fix, other change,
dev setup
---
 src/gluonts/nursery/daf/requirements.txt               | 2 +-
 src/gluonts/nursery/few_shot_prediction/pyproject.toml | 2 +-
 src/gluonts/nursery/tsbench/pyproject.toml             | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/gluonts/nursery/daf/requirements.txt b/src/gluonts/nursery/daf/requirements.txt
index 7d778d2960..f6716b438a 100644
--- a/src/gluonts/nursery/daf/requirements.txt
+++ b/src/gluonts/nursery/daf/requirements.txt
@@ -2,6 +2,6 @@ torch==2.2.0
 tensorboard==2.3.0
 numpy==1.22.0
 pandas==1.1.5
-scikit-learn~=1.4.0
+scikit-learn~=1.5.0
 scipy==1.10.0
 matplotlib==3.3.2
\ No newline at end of file
diff --git a/src/gluonts/nursery/few_shot_prediction/pyproject.toml b/src/gluonts/nursery/few_shot_prediction/pyproject.toml
index 2b61158b4d..5f48a051eb 100644
--- a/src/gluonts/nursery/few_shot_prediction/pyproject.toml
+++ b/src/gluonts/nursery/few_shot_prediction/pyproject.toml
@@ -18,7 +18,7 @@ python = "^3.8,<3.10"
 pytorch-lightning = "^1.4.4"
 sagemaker = "^2.218.0"
 scikit-learn = "^1.4.0"
-torch = "^1.9.0"
+torch = "^2.2.0"
 sagemaker-training = "^3.9.2"
 python-dotenv = "^0.19.0"
 xlrd = "^2.0.1"
diff --git a/src/gluonts/nursery/tsbench/pyproject.toml b/src/gluonts/nursery/tsbench/pyproject.toml
index 767cfe0f2f..9af39de4c2 100644
--- a/src/gluonts/nursery/tsbench/pyproject.toml
+++ b/src/gluonts/nursery/tsbench/pyproject.toml
@@ -37,7 +37,7 @@ scipy = "^1.6.3"
 seaborn = "^0.11.2"
 statsmodels = "^0.13.0"
 toolz = "^0.11.1"
-torch = "^1.10.1"
+torch = "^2.2.0"
 tqdm = "^4.60.0"
 ujson = "^5.1.0"
 xgboost = "^1.4.1"