diff --git a/test/sagemaker_tests/pytorch/training/requirements.txt b/test/sagemaker_tests/pytorch/training/requirements.txt
index 3222d81b84b9..3227c8653154 100644
--- a/test/sagemaker_tests/pytorch/training/requirements.txt
+++ b/test/sagemaker_tests/pytorch/training/requirements.txt
@@ -5,7 +5,7 @@ protobuf
 sagemaker>=2.180.0
 coverage
 flake8==3.7.7
-Flask==2.2.5
+Flask==3.1.3
 mock==2.0.0
 # Preserve pytest caching behavior in earlier versions
 pytest<8.1