diff --git a/src/sample_pytorch_gpu_project/.devcontainer/requirements.txt b/src/sample_pytorch_gpu_project/.devcontainer/requirements.txt
index e450735..81aaa34 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.7.1
 torchvision==0.22.1
 # the below are used by AML, can be deleted if not using AML
 mlflow==3.1.1
-azureml-mlflow==1.60.0
+azureml-mlflow==1.60.0.post1