diff --git a/operations/sagemaker-mlflow-trace-evaluate-langgraph-agent/requirements.txt b/operations/sagemaker-mlflow-trace-evaluate-langgraph-agent/requirements.txt
index a67bc8f..4e13d60 100644
--- a/operations/sagemaker-mlflow-trace-evaluate-langgraph-agent/requirements.txt
+++ b/operations/sagemaker-mlflow-trace-evaluate-langgraph-agent/requirements.txt
@@ -4,7 +4,7 @@ evaluate>=0.4.3
 ipykernel>=6.29.5
 langchain-aws>=0.2.14
 langgraph>=0.3.2
-mlflow==2.20.2
+mlflow==3.5.0rc0
 nltk>=3.9.1
 pandas>=2.2.0
 psutil>=7.0.0