fix(pytorch-mnist): update torch packages (#13284)

Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>
hongbo-miao · Dec 18, 2023 · 4c66410 · 4c66410
1 parent cbc5b10
commit 4c66410
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 46 deletions.
diff --git a/aws/amazon-sagemaker/pytorch-mnist/poetry.lock b/aws/amazon-sagemaker/pytorch-mnist/poetry.lock
diff --git a/aws/amazon-sagemaker/pytorch-mnist/pyproject.toml b/aws/amazon-sagemaker/pytorch-mnist/pyproject.toml
@@ -6,8 +6,8 @@ authors = ["Hongbo Miao"]
 
 [tool.poetry.dependencies]
 python = "3.10.x"
-torch = "2.1.1"
-torchvision = "0.16.1"
+torch = "2.1.2"
+torchvision = "0.16.2"
 
 [tool.poetry.group.dev.dependencies]
 poethepoet = "0.24.4"