diff --git a/src/sample_pytorch_gpu_project/.devcontainer/requirements.txt b/src/sample_pytorch_gpu_project/.devcontainer/requirements.txt
index 3533149..4b0ba8c 100644
--- a/src/sample_pytorch_gpu_project/.devcontainer/requirements.txt
+++ b/src/sample_pytorch_gpu_project/.devcontainer/requirements.txt
@@ -2,4 +2,4 @@ torch==2.10.0
 torchvision==0.25.0
 # the below are used by AML, can be deleted if not using AML
 mlflow==3.9.0
-azureml-mlflow==1.62.0
+azureml-mlflow==1.62.0.post2