diff --git a/sdks/python/container/ml/py312/base_image_requirements.txt b/sdks/python/container/ml/py312/base_image_requirements.txt
index c72558f8acd9..3e2cef4e0243 100644
--- a/sdks/python/container/ml/py312/base_image_requirements.txt
+++ b/sdks/python/container/ml/py312/base_image_requirements.txt
@@ -173,7 +173,7 @@ pymongo==4.16.0
 PyMySQL==1.1.2
 pyparsing==3.3.2
 pyproject_hooks==1.2.0
-pytest==8.4.2
+pytest==9.0.3
 pytest-timeout==2.4.0
 pytest-xdist==3.8.0
 python-dateutil==2.9.0.post0
diff --git a/sdks/python/container/ml/py312/gpu_image_requirements.txt b/sdks/python/container/ml/py312/gpu_image_requirements.txt
index 6dbf6250d598..e5a77fdfba9f 100644
--- a/sdks/python/container/ml/py312/gpu_image_requirements.txt
+++ b/sdks/python/container/ml/py312/gpu_image_requirements.txt
@@ -232,7 +232,7 @@ pymongo==4.16.0
 PyMySQL==1.1.2
 pyparsing==3.3.2
 pyproject_hooks==1.2.0
-pytest==8.4.2
+pytest==9.0.3
 pytest-timeout==2.4.0
 pytest-xdist==3.8.0
 python-dateutil==2.9.0.post0