From 77048778b7663ef8ce50a47d08cc5cafa3b87f9f Mon Sep 17 00:00:00 2001 From: "Jessica Zhang (NY)" Date: Tue, 3 Dec 2024 16:45:29 -0500 Subject: [PATCH] Fix sagemaker/pydantic incompatiblity causing test failures --- dev-requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dev-requirements.txt b/dev-requirements.txt index 163e41e67..50d9a9dbb 100644 --- a/dev-requirements.txt +++ b/dev-requirements.txt @@ -22,7 +22,7 @@ pytest pytest-cov pytorch-lightning==2.3.1 tensorboard==2.14.0 -sagemaker>=2.149.0 +sagemaker=2.232.2 torch-model-archiver>=0.4.2 torch==2.2.1 torchmetrics==0.10.3