diff --git a/test/sagemaker_tests/autogluon/training/requirements.txt b/test/sagemaker_tests/autogluon/training/requirements.txt
index 88a3e6cf5503..ce8f6660466f 100644
--- a/test/sagemaker_tests/autogluon/training/requirements.txt
+++ b/test/sagemaker_tests/autogluon/training/requirements.txt
@@ -3,7 +3,7 @@ protobuf>=3.20,<3.21
 sagemaker>=2.187,<3
 coverage
 flake8==3.7.7
-Flask==1.1.1
+Flask==3.1.3
 mock
 pytest
 pytest-cov