diff --git a/test/sagemaker_tests/pytorch/inference/requirements.txt b/test/sagemaker_tests/pytorch/inference/requirements.txt
index ab8e1c3587a4..52a0cf907ad0 100644
--- a/test/sagemaker_tests/pytorch/inference/requirements.txt
+++ b/test/sagemaker_tests/pytorch/inference/requirements.txt
@@ -3,7 +3,7 @@ coverage
 # Docker v7.0.0 breaks compatibility with Docker Compose v1 (SageMaker Local)
 docker<=6.1.3
 docker-compose
-Flask==1.1.1
+Flask==3.1.3
 fabric
 flake8==3.7.7
 gitpython